跳到正文
原文
Hugging Face Blog·· 2023-04-27AI 评分36

用 🤗 Transformers、TensorFlow 和 TPU 从零训练语言模型

Training a language model with 🤗 Transformers using TensorFlow and TPUs

AI 导读

Hugging Face 发布端到端教程,演示如何用 🤗 Transformers、TensorFlow 和 TPU 从零训练一个 RoBERTa base 模型,涵盖训练 tokenizer、在 WikiText v1 上分词并转为 TFRecord 上传 GCS,再到模型训练与上传的完整流程。代码默认使用 BERT 规模模型,可通过修改配置扩展到更大模型和更强 TPU pod 切片。

来源:Hugging Face Blog · huggingface.co