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