跳到正文
原文
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