Hugging Face Blog·· 2021-12-08精选AI 评分62
从零训练代码生成模型 CodeParrot
Training CodeParrot 🦜 from Scratch
AI 导读
Hugging Face 发布教程,介绍如何从零训练名为 CodeParrot 的 GPT-2 代码生成模型,用于自动补全 Python 代码。作者从 Google BigQuery 的 GitHub dump 中筛选出 180GB、2000 万个 Python 文件,去重并按 Codex 论文的清洗规则处理后得到 50GB 数据集 codeparrot-clean。
推荐理由
完整拆解了从数据清洗、tokenizer 训练到多卡训练循环的每一步,可迁移到自建代码生成模型的流程。
来源:Hugging Face Blog · huggingface.co