Hello! Thanks for the amazing repo!
Below is the training plot on a custom dataset when trained for 5 epochs.

When I try to resume the training from the last checkpoint, the legend is missing from the updated training plot