Solved by qiaoyang_luo in post #3 
              
              
                Yes, I solved this by removing unnecessary keys. 
# There are two kinds of pretrained model
# load torchvision pretrain model
vgg16 = models.vgg16(pretrained=True)
pretrained_dict = vgg16.state_dict()
model_dict = model.state_dict()

# 1. filter out unnecessary keys
pretrained_dict = {k: v for k, v …