Closed kenoharada closed 3 years ago
Add HRSSM model, codes are based on the author's implemetation https://github.com/taesupkim/vta
Original Paper: Variational Temporal Abstraction, Kim et al. https://arxiv.org/pdf/1910.00775.pdf
Original Code https://github.com/taesupkim/vta
Download the dataset Google drive: https://drive.google.com/drive/u/2/folders/1cPdCyrUfHxkidTbkW790H4Jy9jLTonzr
please put this file in HRSSM directory
$ python train.py --seq-size 40 --seg-num 5 --seg-len 16
Add HRSSM model, codes are based on the author's implemetation https://github.com/taesupkim/vta
Hierarchical Recurrent State Space Model(HRSSM)
Original Paper: Variational Temporal Abstraction, Kim et al. https://arxiv.org/pdf/1910.00775.pdf
Original Code https://github.com/taesupkim/vta
3dmaze dataset
Download the dataset Google drive: https://drive.google.com/drive/u/2/folders/1cPdCyrUfHxkidTbkW790H4Jy9jLTonzr
please put this file in HRSSM directory
Run
$ python train.py --seq-size 40 --seg-num 5 --seg-len 16