← 返回资源分享
How to Train BERT with an Academic Budget
论文
论文
发布时间2021-04-15
发表EMNLP 2021 11 · arXiv:2104.07705
作者:Omer Levy,Peter Izsak,Moshe Berchansky
详细介绍
While large language models a la BERT are used ubiquitously in NLP, pretraining them is considered a luxury that only a few well-funded industry labs can afford. How can one train such models with a more modest budget? We present a recipe for pretraining a masked language model in 24 hours using a single low-end deep learning server. We demonstrate that through a combination of software optimizations, design choices, and hyperparameter tuning, it is possible to produce models that are competitive with BERT-base on GLUE tasks at a fraction of the original pretraining cost.
代码仓库 (4)
IntelLabs/academic-budget-bert官方PyTorch
peteriz/academic-budget-bert官方PyTorch
octanove/shibaPyTorch
yxzwang/normalized-information-payloadPyTorch
