Hugging Face Blog·· 2023-01-24AI 评分45
Hugging Face Optimum 集成 ONNX Runtime 实现模型训练加速
Optimum+ONNX Runtime - Easier, Faster training for your Hugging Face models
AI 导读
Hugging Face Optimum 库通过集成 Microsoft ONNX Runtime，为 Transformer 模型提供开源训练加速方案。在单节点 8 GPU A100 环境下，结合 DeepSpeed ZeRO Stage 1，训练速度提升 39% 至 130%，平均提速超 35%。
来源：Hugging Face Blog · huggingface.co