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