pytorch lightning:
自动early stopping,自动batch_size, leaning rate搜索
当模型比较复杂时,为了提高代码的可读性,建议先使用torch.nn.Module构建网络,再使用pytorch_lightning.LightningModule对其进行包装
使用Pytorch进行开发时,到这里模型定义就结束了,其余的训练、验证的具体实现会放在train.py等训练代码中;但Pytorch Lightning模型实现的是整个系统,所以训练的细节也会在这个类中实现
当模型比较复杂时,为了提高代码的可读性,建议先使用torch.nn.Module构建网络,再使用pytorch_lightning.LightningModule对其进行包装