Hi,
I faced similar issue trying to save using .pth. When I saved as .pt it worked fine.
Try this:
Saving the model using
torch.save(my_model.state_dict(),‘somename.pt’)
Load the model using
my_model2 = GraphConv(w2i, p2i, l2i, r2i, s2i, words, pos, lems, 512, 512, 3)
my_model2.load_state_dict(torch.load(‘somename.pt’))
print(test_fn(my_model2))