This package is a PyTorch implementation of the paper Learning from Trajectories via Subgoal Discovery, by Sujoy Paul, Jeroen van Baar and Amit K Roy-Chowdhury, published at NeurIPS 2019.
python main.py --path-to-trajectories <path/to/trajectories/>Here is a link to the AntMaze trajectories. The outputs can also be visualized with vis.py as follows.
python vis.py --path-to-trajectories <path/to/trajectories> --pretrained-ckpt <checkpointname>