Hugging Face Blog·· 2023-01-24AI 评分52
Hugging Face Optimum 集成 ONNX Runtime 训练,模型训练加速最高达 130%
Optimum+ONNX Runtime - Easier, Faster training for your Hugging Face models
AI 导读
Hugging Face 与 Microsoft ONNX Runtime 团队合作,通过 Optimum 库的 ORTTrainer API 为 Transformer 模型训练加速,配合 DeepSpeed ZeRO Stage 1 在单台 8 卡 Nvidia A100 节点上测得 39% 到 130% 的加速。
来源:Hugging Face Blog · huggingface.co