I save my model using the following code.
Path = “Model/model.pth”
torch.save(Network.state_dict(), Path)
Now, i am trying to load the trained model using following code,
model = torch.load(“Model/model.pth”)
model.eval()
But the following error occurred.
AttributeError: ‘collections.OrderedDict’ object has no attribute ‘eval’