Hugging Face Blog·· 2021-12-08精选AI 评分64
Hugging Face 教程:从零开始训练 CodeParrot 代码生成模型
Training CodeParrot 🦜 from Scratch
AI 导读
Hugging Face 发布教程,介绍如何从零训练 110M 和 1.5B 参数的 GPT-2 模型 CodeParrot 以生成 Python 代码。该模型使用基于 180GB Python 代码清洗后的 50GB 数据集训练,并采用 🤗 Accelerate 库在 16 张 A100 GPU 上完成分布式训练。
推荐理由
提供了训练 GPT-2 大模型的具体步骤,涵盖数据清洗、分词器设计和分布式训练,可复用于自研代码模型。
来源:Hugging Face Blog · huggingface.co