Thank you.
Here is the code:
train_model.load_state_dict(
#torch.load(os.path.join(final_output_path, config.model_prefix + str(curr_iter) + ‘.pth’)), resume=True) # this works fine
torch.load(os.path.join(final_output_path, config.model_prefix + str(curr_iter) + ‘.pth’)),strict=False) #this doesnot work and give the error I list above.