Load best model checkpoint

[Pacman1984]

Hi everyone,

is flash loading the best model checkpoint after training for max_epochs (best practice) or does it train max_epochs and holds the last model checkpoint?

Thanks and all the best
Sebastian

[aniketmaurya]

You can use model checkpoint callback and load the best weight.

from pytorch_lightning.callbacks import ModelCheckpoint

checkpoint_callback = ModelCheckpoint(...)
trainer = Trainer(callbacks=[checkpoint_callback])
trainer.fit(model)

checkpoint_callback.best_model_path