PytorchLightning使用说明

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的输入;