Skip to content

Release of GraphMDN

Latest
Compare
Choose a tag to compare
@dchannah dchannah released this 26 Jul 13:14

This release contains the code used in our paper on Graph Mixture Density Networks, currently located on arXiv. In addition to the code itself, the release includes a compressed archive containing a nested directory called checkpoint/trained_model, inside of which are two files:

  1. ckpt_best.pth.tar: The actual checkpoint file
  2. params.txt: The parameters used in the training run used to produce the checkpoint. Evaluating the checkpoint using the instructions in the README will cause this file to be automatically loaded.

To get these files in the correct place, download the archive and extract it in the root folder of the repository (so that the checkpoint folder is accessible from the root).