Update ScoreInfer to load model using the new config

awslabs / dgl-ke

High performance, easy-to-use, and scalable package for learning large-scale knowledge graph embeddings.

https://dglke.dgl.ai/doc/

Apache License 2.0

1.28k stars 196 forks source link

Update ScoreInfer to load model using the new config #191

Closed sebastiandro closed 3 years ago

sebastiandro commented 3 years ago

Description of changes: Running dglke_predict fails because of the updated model config. This change updates the ScoreInfer load_model method to use the correct config keys.

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

classicsong commented 3 years ago

Thanks, Sebastian.