I get this error when I try to load the model the model is getting saved, when I try to load it it’s throwing a error.
The syntax which I used is the one below:
model_scratch.load_state_dict(torch.load('saved_models/model_scratch.pt'))
and the error which I got is this
IncompatibleKeys(missing_keys=[], unexpected_keys=[])