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

[Question] --batch_size_eval #98

Closed HenryYihengXu closed 4 years ago

HenryYihengXu commented 4 years ago

Is it possible to set this flag to all edges? I mean not by specifying the exact number, but some flags like "all_edges."

classicsong commented 4 years ago

This just controls the minibatch size in evaluation to make sure there will not be an OOM problem in evaluation.

HenryYihengXu commented 4 years ago

I see. So you are saying if we omit this flag, by default it uses all edges for evaluation right?

classicsong commented 4 years ago

The evaluation will use all edges in the evaluation set (you can provide your own eval data) no matter what value you set for batch_size_eval.