Describe the bug
Details of the bug can be found in this conversation here but essentially save_last actually saves a copy of the last checkpoint that was saved rather than the checkpoint for the last epoch.
We should change this so we actually do save the last checkpoint, by e.g. either by having a customer saver or just saving out the checkpoint for the current epoch separately to the best one and keep overwriting so we keep the last one (might be done by a custom saver)
Describe the bug
Details of the bug can be found in this conversation here but essentially save_last actually saves a copy of the last checkpoint that was saved rather than the checkpoint for the last epoch.
We should change this so we actually do save the last checkpoint, by e.g. either by having a customer saver or just saving out the checkpoint for the current epoch separately to the best one and keep overwriting so we keep the last one (might be done by a custom saver)