跳到正文
原文
Hugging Face Blog·· 2022-10-21AI 评分48

从 PyTorch DDP 到 Accelerate 再到 Trainer:轻松掌握分布式训练

From PyTorch DDP to Accelerate to Trainer, mastery of distributed training with ease

AI 导读

这是一篇 Hugging Face 教程,讲解如何以三种递进的抽象层级在多 GPU 上进行分布式数据并行(DDP)训练:原生 PyTorch 的 pytorch.distributed 模块、🤗 Accelerate 的轻量封装(代码几乎无需改动即可在单 GPU 和 TPU 上运行),以及 🤗 Transformers 的高层 Trainer API。

来源:Hugging Face Blog · huggingface.co