Releases: vectra-ai-research/GraphMDN
Releases · vectra-ai-research/GraphMDN
Release of GraphMDN
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:
ckpt_best.pth.tar
: The actual checkpoint fileparams.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).