Hello,
I got the following error when I tried to load my model:

RuntimeError: Error(s) in loading state_dict for VGG:


Missing key(s) in state_dict: “classifier.0.weight”, “classifier.0.bias”, “classifier.3.weight”, “classifier.3.bias”, “classifier.6.weight”, “classifier.6.bias”.


Unexpected key(s) in state_dict: “classifier.fc1.weight”, “classifier.fc1.bias”, “classifier.fc2.weight”, “classifier.fc2.bias”.

The code that I used to save and load my model is as follow:
torch.save(model.state_dict(), ‘check_point.pth’)
model.load_state_dict(torch.load(‘check_point.pth’))
What am I missing?