PyTorch Lightning

Orchestra-Research/AI-Research-SKILLs/08-distributed-training/pytorch-lightning

12.1k 0TeX研究
平台通用(跨宿主)

带 Trainer 类、自动分布式训练(DDP/FSDP/DeepSpeed)、回调系统、极少样板的高层 PyTorch 框架。同代码从笔记本扩到超算。想要带内置最佳实践的干净训练循环时使用。

全部 skill