I’m working through the pytorch classifier models (thumbs, facial expressions, etc) and have noticed that when I train and save a model (my_xyz_model.pth), I cannot seem to reload the model and have it work properly at a later time. I can successfully reload the model, and it runs, outputting predictions, but the accuracy is essentially the same as it would be if I ran it using only the transfer model, before training it on my classification data.
The interface seems to indicate that all I have to do is load up my notebook, ensure the path to my trained model is in the model path field, and click load model.
Am I missing a step in this process? Do I need to do something in the code block where I load the transfer learning model (Resnet in my case)?