跳到正文
原文
Hugging Face Blog·· 2021-04-16精選AI 評分63

Hugging Face 推出 Accelerate,簡化 PyTorch 分散式訓練與混合精度支援

Introducing 🤗 Accelerate

AI 導讀

Hugging Face 推出 Accelerate,PyTorch 訓練腳本加入五行程式碼後,即可支援不同分散式設定及混合精度訓練。Accelerate 提供統一 API,準備模型、最佳化器及 DataLoader,並處理裝置配置和反向傳播。

推薦理由

Accelerate 將 PyTorch 分散式訓練及混合精度所需樣板封裝成少量 API 改動,並提供統一啟動器,呈現減少訓練腳本跨環境改動的實作方式。

來源:Hugging Face Blog · huggingface.co