Github repo for the public code of MUSE-VAE
You can download pretrained models for PFSD from PFSD models
Place the unzipped directory under the
ckptsdirectory as follows.
ckpts |- pretrained_models_pfsd
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/pfsddirectory as follows.
datasets |- pfsd
- Raw data and details of PFSD can be found here
- You can use the script
scripts/pfsd/eval.shto get the evaluation results for PFSD reported in the paper.
- You can use the scripts starting with
scripts/pfsd/to train each of the network.
sh train_lg_cvae.sh sh train_sg_net.sh sh train_micro.sh