I’m using PyTorch for training GCN, I have a simple model that saved using this command:
torch.save(MyNet().state_dict(), PATH)

Then I load it by doing the following:
model = MyNet()
model.load_state_dict(torch.load(PATH))
model.eval()

But then when I tried to input the data as follows:
output = model(dataset)

I got this error for the previous line:
TypeError: 'NoneType' object is not callable

I don’t know what’s the reason for that, could you please help me?