Learning Deep Generative Models of Graphs
This is an implementation of Learning Deep Generative Models of Graphs by Yujia Li, Oriol Vinyals, Chris Dyer, Razvan Pascanu, Peter Battaglia.
- Train with batch size 1:
- Train with batch size larger than 1:
90% accuracy for cycles compared with 84% accuracy reported in the original paper.
On AWS p3.2x instance (w/ V100), one epoch takes ~526s for batch size 1 and takes ~238s for batch size 10.
We would like to thank Yujia Li for providing details on the implementation.