Hugging Face Blog·· 2021-04-16精选AI 评分74
Hugging Face 推出开源库 Accelerate
Introducing 🤗 Accelerate
AI 导读
Hugging Face 推出开源库 Accelerate,允许开发者仅需在原生 PyTorch 训练循环中添加数行代码,即可统一运行在多 GPU、TPU 以及混合精度等不同硬件配置上。该库通过统一 API 抽象底层样板代码,自动处理设备放置与数据分发,并提供配置向导与 CLI 启动命令以简化分布式训练脚本的执行。
推荐理由
原文对比了原生PyTorch分布式模板代码与简化改动,读者可以借此评估统一多设备训练脚本的迁移成本。
来源:Hugging Face Blog · huggingface.co