KB2E Train_TransE 训练样本量过大会导致 Segmentation fault，怎么解决？

Train_TransE 训练样本量过大会导致 Segmentation fault，怎么解决？

Open SHK92 opened this issue 4 years ago • 2 comments

我用自己的数据生成的train.txt, entity2id.txt 和 relation2id.txt。train的样本量在1000w级。会报错。train.txt抽一个和FB15k中train.txt样本量相当的sample，是可以正常跑出结果的。但是数据量上去就不得行了。

Jun 02 '20 11:06 SHK92

源代码有一些需要优化的地方，否则很容易造成内存泄漏，或许可以参考一下这个issue： https://github.com/thunlp/KB2E/issues/5

Aug 19 '20 14:08 t170815518

我想问下该给Train_TransE中的主函数传入什么参数

Mar 30 '21 12:03 yaowang1112