[人工智能]【Pytorch Lightning】LightingModule API使用

之前的文章只是大致介绍Lighting使用的相关特性，Lighting主要使用的有LightingModule、Trainer等。下面我们来看看LightingModule.

LightningModule 将您的 PyTorch 代码分为 5 个部分:

• 计算（初始化）
• 训练循环（training_step）
• 验证循环（validation_step）
• 测试循环（test_step）
• 优化器 (config_optimizers)

官网演示视频

需要注意的几件事：

1 本身还是Pytorch代码
2 PyTorch 代码不是被抽象了 - 只是被组织起来 了
3 不在 LightningModule 中的所有其他代码都已由trainer为您自动完成

net = Net()
trainer = Trainer()
trainer.fit(net

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