Skip to content

Releases: vectra-ai-research/GraphMDN

Release of GraphMDN

26 Jul 13:14
Compare
Choose a tag to compare

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).