PytorchLightning使用说明
PytorchLightning 基本使用
hook使用顺序
PytorchLightning的最初使用日常修改只需要自定义其中一部分函数即可,包含但不限于:
- training_epoch_end
- training_step
- training_step_end
- validation_epoch_end
- validation_step
- validation_step_end
在训练过程中,一般都会定义一个类继承自pl.LightningModule,然后重写里面的函数;
training_step函数对batch进行预测,返回一个loss;并将该loss作为training_step_end的输入;