Easiest way to load the best model checkpoint after training w/ pytorch lightning

Hi @tleyden , happy to help. Please review the following resource on model checkpointing and retrieval.

A common flow would be to log a model checkpoint as in the example then to also log a “best model” artifact. Since artifacts are versioned you don’t have to worry about renaming the new “best model” artifact. Then at the end of your run you not only have an artifact history of your model at each of the checkpoints but also a versioned history of all the best models.