intfloat / SimKGC

ACL 2022, SimKGC: Simple Contrastive Knowledge Graph Completion with Pre-trained Language Models
188 stars 36 forks source link

About PyTorch Ver #5

Closed EurusNotes closed 2 years ago

EurusNotes commented 2 years ago

Hi Liang,

What version of PyTorch are you using?

intfloat commented 2 years ago

I use pytorch 1.6 and 1.7 to run the experiments. Any pytorch version >= 1.6 should work.

Please let me know if you encounter any problems.

EurusNotes commented 2 years ago

Thank you for your reply!