├── .gitignore ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── assets ├── 0.png └── 1.png ├── data └── download_dataset.sh ├── docker ├── Dockerfile └── environment.yaml ├── nautilus └── launch.py └── puppeteer ├── __init__.py ├── common ├── __init__.py ├── buffer.py ├── init.py ├── layers.py ├── logger.py ├── math.py ├── mocap_dataset.py ├── parser.py ├── scale.py ├── seed.py └── world_model.py ├── config.yaml ├── envs ├── __init__.py ├── dm_control_wrapper.py ├── tasks │ ├── __init__.py │ ├── arenas.py │ ├── run_through_corridor.py │ └── walk.py ├── tracking.py ├── transfer.py ├── walkers │ ├── assets │ │ └── humanoid_CMU_V2020.xml │ └── cmu_humanoid.py └── wrappers │ ├── __init__.py │ ├── humanoid.py │ ├── tensor.py │ └── time_limit.py ├── evaluate.py ├── tdmpc2.py ├── train.py └── trainer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/.gitignore -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/README.md -------------------------------------------------------------------------------- /assets/0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/assets/0.png -------------------------------------------------------------------------------- /assets/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/assets/1.png -------------------------------------------------------------------------------- /data/download_dataset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/data/download_dataset.sh -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/docker/Dockerfile -------------------------------------------------------------------------------- /docker/environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/docker/environment.yaml -------------------------------------------------------------------------------- /nautilus/launch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/nautilus/launch.py -------------------------------------------------------------------------------- /puppeteer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /puppeteer/common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /puppeteer/common/buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/common/buffer.py -------------------------------------------------------------------------------- /puppeteer/common/init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/common/init.py -------------------------------------------------------------------------------- /puppeteer/common/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/common/layers.py -------------------------------------------------------------------------------- /puppeteer/common/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/common/logger.py -------------------------------------------------------------------------------- /puppeteer/common/math.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/common/math.py -------------------------------------------------------------------------------- /puppeteer/common/mocap_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/common/mocap_dataset.py -------------------------------------------------------------------------------- /puppeteer/common/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/common/parser.py -------------------------------------------------------------------------------- /puppeteer/common/scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/common/scale.py -------------------------------------------------------------------------------- /puppeteer/common/seed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/common/seed.py -------------------------------------------------------------------------------- /puppeteer/common/world_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/common/world_model.py -------------------------------------------------------------------------------- /puppeteer/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/config.yaml -------------------------------------------------------------------------------- /puppeteer/envs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/envs/__init__.py -------------------------------------------------------------------------------- /puppeteer/envs/dm_control_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/envs/dm_control_wrapper.py -------------------------------------------------------------------------------- /puppeteer/envs/tasks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /puppeteer/envs/tasks/arenas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/envs/tasks/arenas.py -------------------------------------------------------------------------------- /puppeteer/envs/tasks/run_through_corridor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/envs/tasks/run_through_corridor.py -------------------------------------------------------------------------------- /puppeteer/envs/tasks/walk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/envs/tasks/walk.py -------------------------------------------------------------------------------- /puppeteer/envs/tracking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/envs/tracking.py -------------------------------------------------------------------------------- /puppeteer/envs/transfer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/envs/transfer.py -------------------------------------------------------------------------------- /puppeteer/envs/walkers/assets/humanoid_CMU_V2020.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/envs/walkers/assets/humanoid_CMU_V2020.xml -------------------------------------------------------------------------------- /puppeteer/envs/walkers/cmu_humanoid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/envs/walkers/cmu_humanoid.py -------------------------------------------------------------------------------- /puppeteer/envs/wrappers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/envs/wrappers/__init__.py -------------------------------------------------------------------------------- /puppeteer/envs/wrappers/humanoid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/envs/wrappers/humanoid.py -------------------------------------------------------------------------------- /puppeteer/envs/wrappers/tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/envs/wrappers/tensor.py -------------------------------------------------------------------------------- /puppeteer/envs/wrappers/time_limit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/envs/wrappers/time_limit.py -------------------------------------------------------------------------------- /puppeteer/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/evaluate.py -------------------------------------------------------------------------------- /puppeteer/tdmpc2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/tdmpc2.py -------------------------------------------------------------------------------- /puppeteer/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/train.py -------------------------------------------------------------------------------- /puppeteer/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nicklashansen/puppeteer/HEAD/puppeteer/trainer.py --------------------------------------------------------------------------------