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