How exactly are you saving the model? When to save the model it usually saved as an OrderedDict and hence it’ll be a dictionary that contains multiple objects which can be obtained via a given key. For example, I save my model like,
torch.save({'epoch':epoch,
            'loss': loss,
            'model_state_dict':model.state_dict(), 
            'optim_state_dict':optim.state_dict()}, "path/to/location.pt')

then to load I do,
state_dict = torch.load(f="path/to/location.pt", map_location=device)
epoch = state_dict['epoch']
loss = state_dict['loss']
model.load_state_dict(state_dict['model_state_dict'])
optim.load_state_dict(state_dict['optim_state_dict'])

Could you print out model from the line model = torch.load("model.pt") ?