/deep_mind_model

this is a graph model based on the famous openai work

Primary LanguagePython

DeepMind Graph Model

This is a pytorch implementation of Deep Mind's graph net. You need pytorch and pytorch_geometric installed for the code to run.

The shortest path benchmark is implemented in shortest_path.py.