Transfer Learning Missing key(s) in state_dict & Unexpected key(s) in state_dict

[ptrblck]

Your issue is probably related to this.