dataloader for sdd

[Zhiwei (Danny) Xu]

Hi, Do you have the dataloader for SDD? For CSM and TiledMP, there are a single .npz for each dataset; but for SDD, there are 8 types each type has some "video" subfolder where each subfolder has several .npz. So I am wondering if you have the dataloader for SDD for reproduce? Thanks.

[Ryo Yonetani]

Hi! Please refer to the icml2021 branch.

neural-astar/planning_experiment/utils/SDD.py

Line 12 in ff7ef16

 class SDD(Dataset): 

neural-astar/scripts/train_sdd.py

Line 72 in ff7ef16

 train_dataset = SDD(args.data_dir, 

[Zhiwei (Danny) Xu]

Thank you very much.