使用 PyTorch Fully Sharded Data Parallel 加速大模型训练
Accelerate Large Model Training using PyTorch Fully Sharded Data Parallel
精选理由
帮助读者掌握用 PyTorch FSDP 高效训练大模型的方法,降低显存与通信开销。
Accelerate Large Model Training using PyTorch Fully Sharded Data Parallel
帮助读者掌握用 PyTorch FSDP 高效训练大模型的方法,降低显存与通信开销。