Save best checkpoint during a training run #119
Description
Currently we save the final checkpoint as a default, as well as any other frequency of checkpoints that the user provides in the configs (Lightning handles this through save_every_n_epochs). Propose adding logic to save the checkpoint with the best metric value, and tag it with "best"