├── .gitignore ├── CONTRIBUTING.md ├── README.md ├── assets ├── 0.gif ├── 1.gif ├── 2.gif └── 3.gif ├── demo3 ├── __init__.py ├── common │ ├── __init__.py │ ├── discriminator.py │ ├── init.py │ ├── layers.py │ ├── logger.py │ ├── math.py │ ├── parser.py │ ├── scale.py │ ├── seed.py │ ├── trajectory_saver.py │ └── world_model.py ├── config │ ├── demo3.yaml │ ├── eval.yaml │ └── tdmpc2.yaml ├── envs │ ├── __init__.py │ ├── maniskill.py │ ├── metaworld.py │ ├── robosuite.py │ ├── tasks │ │ ├── maniskill_stages.py │ │ ├── mw_stages.py │ │ └── robosuite_stages.py │ ├── utils.py │ └── wrappers │ │ ├── __init__.py │ │ ├── tensor.py │ │ └── vectorized.py ├── evaluate.py ├── storage │ ├── buffer.py │ ├── data_utils.py │ ├── demo3_buffer.py │ └── ensemble_buffer.py ├── tdmpc2.py ├── train.py └── trainer │ ├── __init__.py │ ├── base.py │ ├── demo3_trainer.py │ ├── modem_trainer.py │ └── online_trainer.py ├── docker ├── Dockerfile ├── environment.yaml ├── nvidia_icd.json └── robosuite.yaml ├── nautilus └── launch.py └── results ├── __init__.py └── results_to_csv.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/.gitignore -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/README.md -------------------------------------------------------------------------------- /assets/0.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/assets/0.gif -------------------------------------------------------------------------------- /assets/1.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/assets/1.gif -------------------------------------------------------------------------------- /assets/2.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/assets/2.gif -------------------------------------------------------------------------------- /assets/3.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/assets/3.gif -------------------------------------------------------------------------------- /demo3/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /demo3/common/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/common/__init__.py -------------------------------------------------------------------------------- /demo3/common/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/common/discriminator.py -------------------------------------------------------------------------------- /demo3/common/init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/common/init.py -------------------------------------------------------------------------------- /demo3/common/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/common/layers.py -------------------------------------------------------------------------------- /demo3/common/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/common/logger.py -------------------------------------------------------------------------------- /demo3/common/math.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/common/math.py -------------------------------------------------------------------------------- /demo3/common/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/common/parser.py -------------------------------------------------------------------------------- /demo3/common/scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/common/scale.py -------------------------------------------------------------------------------- /demo3/common/seed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/common/seed.py -------------------------------------------------------------------------------- /demo3/common/trajectory_saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/common/trajectory_saver.py -------------------------------------------------------------------------------- /demo3/common/world_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/common/world_model.py -------------------------------------------------------------------------------- /demo3/config/demo3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/config/demo3.yaml -------------------------------------------------------------------------------- /demo3/config/eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/config/eval.yaml -------------------------------------------------------------------------------- /demo3/config/tdmpc2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/config/tdmpc2.yaml -------------------------------------------------------------------------------- /demo3/envs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/envs/__init__.py -------------------------------------------------------------------------------- /demo3/envs/maniskill.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/envs/maniskill.py -------------------------------------------------------------------------------- /demo3/envs/metaworld.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/envs/metaworld.py -------------------------------------------------------------------------------- /demo3/envs/robosuite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/envs/robosuite.py -------------------------------------------------------------------------------- /demo3/envs/tasks/maniskill_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/envs/tasks/maniskill_stages.py -------------------------------------------------------------------------------- /demo3/envs/tasks/mw_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/envs/tasks/mw_stages.py -------------------------------------------------------------------------------- /demo3/envs/tasks/robosuite_stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/envs/tasks/robosuite_stages.py -------------------------------------------------------------------------------- /demo3/envs/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/envs/utils.py -------------------------------------------------------------------------------- /demo3/envs/wrappers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /demo3/envs/wrappers/tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/envs/wrappers/tensor.py -------------------------------------------------------------------------------- /demo3/envs/wrappers/vectorized.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/envs/wrappers/vectorized.py -------------------------------------------------------------------------------- /demo3/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/evaluate.py -------------------------------------------------------------------------------- /demo3/storage/buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/storage/buffer.py -------------------------------------------------------------------------------- /demo3/storage/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/storage/data_utils.py -------------------------------------------------------------------------------- /demo3/storage/demo3_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/storage/demo3_buffer.py -------------------------------------------------------------------------------- /demo3/storage/ensemble_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/storage/ensemble_buffer.py -------------------------------------------------------------------------------- /demo3/tdmpc2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/tdmpc2.py -------------------------------------------------------------------------------- /demo3/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/train.py -------------------------------------------------------------------------------- /demo3/trainer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /demo3/trainer/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/trainer/base.py -------------------------------------------------------------------------------- /demo3/trainer/demo3_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/trainer/demo3_trainer.py -------------------------------------------------------------------------------- /demo3/trainer/modem_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/trainer/modem_trainer.py -------------------------------------------------------------------------------- /demo3/trainer/online_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/demo3/trainer/online_trainer.py -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/docker/Dockerfile -------------------------------------------------------------------------------- /docker/environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/docker/environment.yaml -------------------------------------------------------------------------------- /docker/nvidia_icd.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/docker/nvidia_icd.json -------------------------------------------------------------------------------- /docker/robosuite.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/docker/robosuite.yaml -------------------------------------------------------------------------------- /nautilus/launch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/nautilus/launch.py -------------------------------------------------------------------------------- /results/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/results/__init__.py -------------------------------------------------------------------------------- /results/results_to_csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/adrialopezescoriza/demo3/HEAD/results/results_to_csv.py --------------------------------------------------------------------------------