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

Hugging Face 发布 🤗 Accelerate 库以简化 PyTorch 分布式训练

Introducing 🤗 Accelerate

AI 导读

Hugging Face 推出开源库 🤗 Accelerate,旨在让 PyTorch 用户无需重写样板代码即可在多种设备配置上运行原始训练脚本。该库提供统一 API 和启动器命令,支持多 GPU、TPU 及混合精度训练,并计划集成 fairscale、DeepSpeed 和 AWS SageMaker。用户只需添加少量代码行即可自动处理设备放置和数据加载器包装,同时兼容传统启动方式。

来源:Hugging Face Blog · huggingface.co