Skip to content

Moved code to lightning

Daniel CARRON requested to merge move-to-lightning into main

Updating codebase to make use of pytorch-lightning

  • Main train/valid loops are handled by lightning
  • Removed device-handling code
  • Each model implements its own train/valid/predict steps
  • Updated config files for use with lightning
  • Using lightning checkpointing and logging callbacks
  • Updated DeviceMonitor to be compatible with lightning

Merge request reports