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