How can one save a gnn model?

[DogitoErgoSum]

The title

[rusty1s]

You can either save the full model

torch.save(my_model, 'model.pt')

or just its state dict/parameters (recommended):

torch.save(my_model.state_dict(), 'model.pt')

[DogitoErgoSum]

And the loading?

[rusty1s]

Either

model = torch.load('model.pt')

or

model = Model()
model.load_state_dict(torch.load('model.pt'))

[DogitoErgoSum]

Ty so much