JinheonBaek / RGCN

Pytorch implementation of a RGCN Link Prediction Model
229 stars 41 forks source link

best_mrr_model.pth #11

Closed zqyJason closed 2 years ago

zqyJason commented 2 years ago

__where can I find the 'best_mrr_model.pth'?

eric2213 commented 1 year ago

Hi! I got the same problem,

FileNotFoundError: [Errno 2] No such file or directory: 'best_mrr_model.pth'

May I ask how did you solve it?

linxiang1911 commented 1 year ago

你们最后是怎么解决的,我也遇到这不到这个路径

JinheonBaek commented 1 year ago

Hi all, to obtain this file: best_mrr_model.pth, you must do train first and then save the checkpoint. See this line: https://github.com/JinheonBaek/RGCN/blob/master/main.py#L84