How to load saved model?

HazyResearch / hgcn

Hyperbolic Graph Convolutional Networks in PyTorch.

603 stars 109 forks source link

How to load saved model? #13

Closed empty-id closed 4 years ago

empty-id commented 4 years ago

Hi, I notice that the script only provides save arg but no load arg.

ines-chami commented 4 years ago

Hi, in our experiments we only needed to reload pretrained embeddings (with the option --pretrained-embeddings). If you would like to reload the entire model weights, feel free to send us a pull request:)