I have written two small models, the first one containing the conv layers and the second containing the fully connected layers with help but while transferring the weights there is an error :
KeyError : ‘unexpected key “0.weight” in state_dict’