jingjing:

num_batches_tracked


This is newly added, so it is in your new state_dict, but not in the downloaded one. We have code to handle this if you are just using load_state_dict on the downloaded one. However, since you are looking keys up in the old state_dict basing on the new state_dict, it won’t work. You could safely skip those.