Error In Calculating Neural Network Test Accuracy

I tried to train my neural network, and then evaluate it's testing accuracy. I am using the code at the bottom of this post to train. The fact is that for other neural networks, I

Solution 1:

You are trying to load a state dict that belongs to another model.

The error shows that your model is the class AlexNet.

RunTimeError:Error(s) in loading state_dict for AlexNet:

But the state dict you are trying to load is from the VGG you posted, which doesn't have the same modules as AlexNet.

You need to use the same model whose state dict you saved before.