intfloat / SimKGC

ACL 2022, SimKGC: Simple Contrastive Knowledge Graph Completion with Pre-trained Language Models

188 stars 36 forks source link

关于模型性能极度提升的问题 #47

Open hcfun opened 2 months ago

hcfun commented 2 months ago

您好 hr_vector+=tail_vector

DataParallel only support tensor/dict

return {'hr_vector': hr_vector, 'tail_vector': tail_vector, 'head_vector': head_vector} 为什么我在获取预训练表示的后面加上hr_vector+=tail_vector这一句之后，模型的性能直接到了100%呢 MR为1 MRR为0.99. 希望能得到您的解答！！非常感谢。

Yu-Chunmiao commented 1 week ago

相当于你在训练的过程中就告诉模型tail 是什么了