├── .gitignore ├── .gitmodules ├── README.md ├── base_agent.py ├── config.yaml ├── envs ├── mujoco_env.py ├── pointmass │ ├── assets │ │ └── pointmass_prob.xml │ └── pointmass_prob_env.py └── sparse_metaworld.py ├── eval.py ├── eval_episodes.py ├── gencmd.py ├── latco.py ├── planners ├── dreamer.py ├── gn_solver.py ├── image_colloc.py ├── latco_gd.py ├── probabilistic_latco.py ├── random.py ├── shooting_cem.py ├── shooting_gd.py ├── shooting_gn.py └── shooting_ilqr.py ├── planning_agent.py ├── plot.py ├── train.py └── utils ├── logging.py ├── models.py ├── tools.py └── wrappers.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/.gitmodules -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/README.md -------------------------------------------------------------------------------- /base_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/base_agent.py -------------------------------------------------------------------------------- /config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/config.yaml -------------------------------------------------------------------------------- /envs/mujoco_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/envs/mujoco_env.py -------------------------------------------------------------------------------- /envs/pointmass/assets/pointmass_prob.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/envs/pointmass/assets/pointmass_prob.xml -------------------------------------------------------------------------------- /envs/pointmass/pointmass_prob_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/envs/pointmass/pointmass_prob_env.py -------------------------------------------------------------------------------- /envs/sparse_metaworld.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/envs/sparse_metaworld.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/eval.py -------------------------------------------------------------------------------- /eval_episodes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/eval_episodes.py -------------------------------------------------------------------------------- /gencmd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/gencmd.py -------------------------------------------------------------------------------- /latco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/latco.py -------------------------------------------------------------------------------- /planners/dreamer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/planners/dreamer.py -------------------------------------------------------------------------------- /planners/gn_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/planners/gn_solver.py -------------------------------------------------------------------------------- /planners/image_colloc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/planners/image_colloc.py -------------------------------------------------------------------------------- /planners/latco_gd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/planners/latco_gd.py -------------------------------------------------------------------------------- /planners/probabilistic_latco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/planners/probabilistic_latco.py -------------------------------------------------------------------------------- /planners/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/planners/random.py -------------------------------------------------------------------------------- /planners/shooting_cem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/planners/shooting_cem.py -------------------------------------------------------------------------------- /planners/shooting_gd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/planners/shooting_gd.py -------------------------------------------------------------------------------- /planners/shooting_gn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/planners/shooting_gn.py -------------------------------------------------------------------------------- /planners/shooting_ilqr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/planners/shooting_ilqr.py -------------------------------------------------------------------------------- /planning_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/planning_agent.py -------------------------------------------------------------------------------- /plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/plot.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/train.py -------------------------------------------------------------------------------- /utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/utils/logging.py -------------------------------------------------------------------------------- /utils/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/utils/models.py -------------------------------------------------------------------------------- /utils/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/utils/tools.py -------------------------------------------------------------------------------- /utils/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zchuning/latco/HEAD/utils/wrappers.py --------------------------------------------------------------------------------