Train_TransE中bfgs()有点问题

thunlp / KB2E

Knowledge Graph Embeddings including TransE, TransH, TransR and PTransE

MIT License

1.4k stars 449 forks source link

Closed chenbjin closed 7 years ago

chenbjin commented 7 years ago

bfgs()中保存实体和关系词向量的代码应该在epoch循环之外吧？

目前这样重复写，并没有保存中间结果，反而延长了训练时间

Mrlyk423 commented 7 years ago

那个是用于训练中间可以随时检测中间结果的，如不需要，可以移到外面。

chenbjin commented 7 years ago

@Mrlyk423 明白了，谢谢！我在此基础上扩展了一下。