Add indication of best checkpoint on plots

The generated plots of the loss should have an indication of where the best epoch is, for example with a vertical line or a point.