跳到正文
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