Anthony_Sirico:

torch.save(model.state_dict(), '/saved_model/test.bin')


Saves the model.state_dict() (which is best practice).

It saves to the file, but then when i try to load the file in a python console, it does not retain the class name:


model = torch.load(‘saved_model/test.bin’)

Loads the saved state dict and assigns it to model.


Then,
mode.eval()
and it states there is not eval.


This is because you only ave the state dict.
You likely want
model = Classifier()
sd = torch.load(...)
model.load_state_dict(sd)
model.eval()

I make this kind of error (that an object isn’t quite what the user thinks it is), too. When things go wrong, I tend to print things a lot to see if they look like what I expect.
Best regards
Thomas