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

Hugging Face 发布 Accelerate 库,让原始 PyTorch 训练脚本适配任意设备

Introducing 🤗 Accelerate

AI 导读

Hugging Face 发布开源库 Accelerate,让原始 PyTorch 训练脚本无需重写即可在任意设备上运行。只需在标准训练脚本中加入约五行代码,即可支持单机或多机多 GPU、TPU 以及混合精度训练,并自动处理设备放置。

推荐理由

官方给出五处代码改动即可跑分布式与混合精度的对比,便于判断迁移成本。

来源:Hugging Face Blog · huggingface.co