Trouble Saving and loading an pretrained Inception Model after Finetuning


I am trying to save my trained model after i have finetuned the pretrained Inception using my dataset.I am able to successfully save the model.When i run the model on my test dataset after the training it works fine.However when i load the model in a separate file and evaluate it on the same test data I am getting a constant target label.

I have used,filepath) and,filepath) to try to save the model

model.load_state_dict(torch.load(filepath).state_dict()) to load the model

I have also changed the model to eval() before testing so that is not the problem

Is there a different method to save for inception net model?

I managed to solve this by saving the entire model and loading the entire model as,filepath)