place optim state_dict loading inside load_checkpoint
#3081
This run and associated checks have been archived and are scheduled for deletion.
Learn more about checks retention