Code

Code

Github repo for the public code of MUSE-VAE

Github

Pretrained Models

  • You can download pretrained models for PFSD from PFSD models

  • Place the unzipped directory under the ckpts directory as follows.

    ckpts
  |- pretrained_models_pfsd

Datasets

  • The pre-processed version of our new dataset, PathFinding Simulation Dataset (PFSD), is available for download at Preprocessed PFSD

  • Place the unzipped pkl files under the datasets/pfsd directory as follows. 
    datasets
  |- pfsd
  • Raw data and details of PFSD can be found here

Running models

  • You can use the script scripts/pfsd/eval.sh to get the evaluation results for PFSD reported in the paper. 
    sh eval.sh
  • You can use the scripts starting with train under scripts/pfsd/ to train each of the network. 
    sh train_lg_cvae.sh
sh train_sg_net.sh
sh train_micro.sh
« Dataset