migalkin / StarE

EMNLP 2020: Message Passing for Hyper-Relational Knowledge Graphs
MIT License
85 stars 17 forks source link

I am wondering how can I save the model and load it? #12

Closed 2258324319 closed 10 months ago

2258324319 commented 1 year ago

the structure of the code is too complex, I think maybe the "SAVE" in the config can help me to save the model, but I don't know how to load it for other test next time.

geraltofrivia commented 1 year ago

Hi, that's right the SAVE parameter when turned on triggers this code block here.

So could you check your working directory for some folder like ./models/{config['DATASET']}/{config['MODEL_NAME']}?