Closed unmeshvrije closed 4 years ago
@chenweize1998 , Sorry to tag you, but would you be kind enough to take a look at this one ?
Which branch of code are you using? Seems like you mixed the code of tensorflow branch and pytorch branch.
I forked from the master branch (or Pytorch branch) on 9 Nov 2019.
Can you kindly give me pointers on how to resolve this and use OpenKE to get HolE embeddings ?
I think HolE.py in PyTorch branch still has the tensorflow code
I have implemented the code for HolE with Pytorch.
I am executing the program using
python train_model.py --gpu --db fb15k237 --mode train --model hole
Note that I added the missing lines in
openke/module/model/__init__.py
to export HolE class.Following is my code i.e.
cat train_model.py
which is in the home directoryOpenKE
Following is my Error log. Is using NegativeSampling bad ?