Hugging Face Blog·· 2021-04-16AI 评分55
Hugging Face 发布开源库 Accelerate,五行代码让 PyTorch 训练脚本支持分布式与混合精度
Introducing 🤗 Accelerate
AI 导读
Hugging Face 发布开源库 Accelerate,让 PyTorch 用户在保留自定义训练循环的前提下,只需向标准脚本加入五行代码即可在多 GPU、多节点、TPU 等分布式环境以及混合精度下运行。
来源:Hugging Face Blog · huggingface.co