search

zjukg / AnKGE

[Paper][AAAI2023] Analogical Inference Enhanced Knowledge Graph Embedding
Apache License 2.0
10 stars 4 forks source link

cuda溢出 #2

Closed passermyh closed 1 year ago

passermyh commented 1 year ago

QQ截图20230418232953 您好，非常感谢您的工作。 我在复现您的代码的时候，使用hake作为base时，我将batchsize已经调整到非常小了，但还是报了溢出的错误，请问哪里还能尽可能不影响结果的改动呢？（12G）

Modberge commented 1 year ago

这里显存溢出的原因是在计算得分函数的kNN最近邻时，我们会将所有实体\关系作为候选计算其得分并进行排序，你可以尝试降低EVAL_BS的值来减小批次大小，这样改动理论上不会对模型效果产生影响，但是需要注意的是，这会极大延长kNN检索时间和模型的测试时间。