Load model, start_epoch

On the “https://github.com/facebookresearch/fastMRI/blob/master/models/unet/train_unet.py”, it says that
if args.resume:
checkpoint, model, optimizer = load_model(args.checkpoint)
args = checkpoint[‘args’]
best_dev_loss = checkpoint[‘best_dev_loss’]
start_epoch = checkpoint[‘epoch’]
del checkpoint

However, I think start_epoch=checkpoint[‘epoch’]+1 is correct because the last epoch is repeated when loading the model in the original code. Especially, when checkpoint[‘epoch’]=args.lr_step_size, scheduler will decay lr twice.

I would appreciate it if you would check this.

1 Like

This looks correct. Thanks for finding the problem. Can you please fix the code?