Skip to content

Latest commit

 

History

History

pytorch_demo

Interactive KGE in PyTorch

Gradient

An interactive training and evaluation demo of Knowledge Graph Embedding (KGE) model training on IPU in PyTorch.

This demo is designed to be run interactively using an IPU notebook service. Please see KgeModelling.ipynb for a complete walkthrough.

Note: requires Poplar SDK >= 3.1.


Included code is licensed under the MIT license. Supporting data files are licensed as detailed in the accompanying LICENSE.md.