跳到正文
原文
Hugging Face Blog·· 2021-04-16精选AI 评分60

Hugging Face 发布 Accelerate 库,让原生 PyTorch 训练脚本支持分布式与混合精度

Introducing 🤗 Accelerate

AI 导读

Hugging Face 发布开源库 Accelerate,让 PyTorch 用户在不改写训练循环的前提下使用分布式训练和混合精度。只需在标准训练脚本中加入约五行代码,Accelerate 就会自动完成设备放置、模型与优化器包装、数据加载器分片以及反向传播处理,并配套 accelerate config 与 accelerate launch 命令行工具。

推荐理由

原文给出五处代码改动即可让原生 PyTorch 脚本支持分布式与混合精度,可据此评估迁移成本。

来源:Hugging Face Blog · huggingface.co