Epoch end checkpoint restarts previous epoch
🐛 Bug
If restarting the training and reloading the model, the epoch that the checkpoint had just completed is restarted rather than beginning the next.
Expected behavior
When a checkpoint upon epoch end is saved, restarting it should resume its state and start the next epoch.