Skip to content

Checkpointing options

Mikael Jacquemont requested to merge checkpoint into master

More flexibility in model checkpointing. Instead of only defining the epoch interval for creating checkpoints, we can use all the parameters of PyTorch Lightning ModelCheckpoint callback.

Merge request reports