fix type error in eq() call

ibalazevic / TuckER

TuckER: Tensor Factorization for Knowledge Graph Completion

MIT License

350 stars 60 forks source link

Closed zenogantner closed 5 years ago

zenogantner commented 5 years ago

At least for my setup (torch 1.0.1, numpy 1.15.4) this is needed.

ibalazevic commented 5 years ago

Sorry, but the code is currently compatible with PyTorch 0.4.0 (see Requirements). I will hopefully try to move it over to 1.0.1 soon.