书写完模型之后,下面就需要书写模型训练的逻辑了。
将 PyTorch 代码组织到 LightningModule 后,Trainer 会自动执行其他所有操作。
官方视频
这种抽象实现了以下目标:
- 您可以通过 PyTorch 代码保持对所有方面的控制,而无需添加抽象。
- Trainer使用来自顶级 AI 实验室(如 Facebook AI Research、NYU、MIT、Stanford 等)的贡献者和用户嵌入的最佳实践……
- Trainner允许覆盖您不希望自动化的任何关键部分。
基本使用
model = MyLightningModule()
trainer = Trainer()
trainer.fit(model, train_dataloader, val_d