[人工智能]【Pytorch Lightning】Trainer API使用（一）

书写完模型之后，下面就需要书写模型训练的逻辑了。

将 PyTorch 代码组织到 LightningModule 后，Trainer 会自动执行其他所有操作。
官方视频

这种抽象实现了以下目标：

• 您可以通过 PyTorch 代码保持对所有方面的控制，而无需添加抽象。
• Trainer使用来自顶级 AI 实验室（如 Facebook AI Research、NYU、MIT、Stanford 等）的贡献者和用户嵌入的最佳实践……
• Trainner允许覆盖您不希望自动化的任何关键部分。

基本使用

model = MyLightningModule()

trainer = Trainer()
trainer.fit(model, train_dataloader, val_d

了解本专栏 订阅专栏 解锁全文 学习会员免费看 赠一年