Hugging Face BlogApr 16, 2021, 12:00 AM重要 78

Introducing 🤗 Accelerate

Hugging Face 推出 Accelerate 庫,只需修改幾行代碼即可讓 PyTorch 訓練無縫適應多 GPU、TPU 與混合精度。

Hugging Face 推出全新開源庫 Accelerate,旨在簡化 PyTorch 的分佈式訓練流程。開發者無需手動處理複雜的設備分配、混合精度(FP16)縮放或分佈式啟動指令。只需使用 Accelerator 類別包裝模型與數據,即可在單機、多 GPU 或 TPU 環境下直接運行相同的代碼,極大提升開發與部署效率。

想看英文原文 / 完整內容?

前往 Hugging Face Blog 原文 →

摘要由 AI 整理,以原文為準。