├── .gitignore ├── .pre-commit-config.yaml ├── .ruff.toml ├── LICENSE ├── MANIFEST.in ├── README.md ├── figures └── teasure_figure.png ├── reds_rl ├── __init__.py ├── clip │ ├── README.md │ ├── __init__.py │ ├── download.py │ ├── layers.py │ ├── model.py │ ├── requirements.txt │ └── tokenizer.py ├── configs │ ├── dreamerv3 │ │ └── configs.yaml │ ├── videogpt │ │ ├── atari.yaml │ │ ├── dmc.yaml │ │ ├── furniturebench.yaml │ │ ├── furnituresim.yaml │ │ ├── metaworld.yaml │ │ └── rlbench.yaml │ ├── vqdiffusion │ │ └── metaworld.yaml │ └── vqgan │ │ ├── atari.yaml │ │ ├── dmc.yaml │ │ ├── furniturebench.yaml │ │ ├── furnituresim.yaml │ │ ├── metaworld.yaml │ │ └── rlbench.yaml ├── dreamerv3 │ ├── Dockerfile │ ├── __init__.py │ ├── agent.py │ ├── behaviors.py │ ├── embodied │ │ ├── __init__.py │ │ ├── core │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── basics.py │ │ │ ├── batch.py │ │ │ ├── batcher.py │ │ │ ├── checkpoint.py │ │ │ ├── config.py │ │ │ ├── counter.py │ │ │ ├── distr.py │ │ │ ├── driver.py │ │ │ ├── flags.py │ │ │ ├── logger.py │ │ │ ├── metaworld_utils.py │ │ │ ├── metrics.py │ │ │ ├── parallel.py │ │ │ ├── path.py │ │ │ ├── random.py │ │ │ ├── rlbench_utils.py │ │ │ ├── space.py │ │ │ ├── timer.py │ │ │ ├── usage.py │ │ │ ├── uuid.py │ │ │ ├── when.py │ │ │ ├── worker.py │ │ │ └── wrappers.py │ │ ├── envs │ │ │ ├── atari.py │ │ │ ├── cliport.py │ │ │ ├── crafter.py │ │ │ ├── dmc.py │ │ │ ├── dmcmulticam.py │ │ │ ├── dmlab.py │ │ │ ├── dummy.py │ │ │ ├── factorworld.py │ │ │ ├── from_dm.py │ │ │ ├── from_gym.py │ │ │ ├── humanoidmulticam.py │ │ │ ├── humanoidmulticam.xml │ │ │ ├── kitchen.py │ │ │ ├── loconav.py │ │ │ ├── loconav_quadruped.py │ │ │ ├── loconav_quadruped.xml │ │ │ ├── metaworld.py │ │ │ ├── minecraft.py │ │ │ ├── minecraft_base.py │ │ │ ├── minecraft_minerl.py │ │ │ ├── pinpad.py │ │ │ ├── rlbench.py │ │ │ └── robodesk.py │ │ ├── replay │ │ │ ├── __init__.py │ │ │ ├── chunk.py │ │ │ ├── generic.py │ │ │ ├── limiters.py │ │ │ ├── naive_chunks.py │ │ │ ├── replays.py │ │ │ ├── reverb.py │ │ │ ├── saver.py │ │ │ └── selectors.py │ │ ├── run │ │ │ ├── __init__.py │ │ │ ├── eval_only.py │ │ │ ├── eval_only_save.py │ │ │ ├── parallel.py │ │ │ ├── train.py │ │ │ ├── train_amp.py │ │ │ ├── train_eval.py │ │ │ ├── train_holdout.py │ │ │ └── train_save.py │ │ └── scripts │ │ │ ├── install-atari.sh │ │ │ ├── install-dmlab.sh │ │ │ ├── install-minecraft.sh │ │ │ ├── plot.py │ │ │ └── xvfb_run.sh │ ├── expl.py │ ├── jaxagent.py │ ├── jaxutils.py │ ├── nets.py │ └── ninjax.py └── videogpt │ ├── __init__.py │ ├── data.py │ ├── loss_vqgan.py │ ├── models │ ├── __init__.py │ ├── stylegan_disc.py │ ├── transformer.py │ ├── videogpt.py │ └── vqgan.py │ ├── reward_models │ ├── __init__.py │ ├── videogpt_reward_model.py │ └── vqdiffusion_reward_model.py │ ├── sampler.py │ └── train_utils.py ├── reds_rl_data ├── __init__.py └── download │ ├── __main__.py │ ├── checkpoint_atari.sh │ ├── checkpoint_dmc.sh │ ├── dataset_atari.sh │ └── dataset_dmc.sh ├── requirements.txt ├── scripts ├── collect_metaworld_data.py ├── collect_rlbench_data.py ├── train_dreamer.py ├── train_reds_metaworld.sh ├── train_reds_rlbench.sh ├── train_videogpt.py └── train_vqgan.py └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.ruff.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/.ruff.toml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- 1 | include requirements.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/README.md -------------------------------------------------------------------------------- /figures/teasure_figure.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/figures/teasure_figure.png -------------------------------------------------------------------------------- /reds_rl/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/__init__.py -------------------------------------------------------------------------------- /reds_rl/clip/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/clip/README.md -------------------------------------------------------------------------------- /reds_rl/clip/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /reds_rl/clip/download.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/clip/download.py -------------------------------------------------------------------------------- /reds_rl/clip/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/clip/layers.py -------------------------------------------------------------------------------- /reds_rl/clip/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/clip/model.py -------------------------------------------------------------------------------- /reds_rl/clip/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/clip/requirements.txt -------------------------------------------------------------------------------- /reds_rl/clip/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/clip/tokenizer.py -------------------------------------------------------------------------------- /reds_rl/configs/dreamerv3/configs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/configs/dreamerv3/configs.yaml -------------------------------------------------------------------------------- /reds_rl/configs/videogpt/atari.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/configs/videogpt/atari.yaml -------------------------------------------------------------------------------- /reds_rl/configs/videogpt/dmc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/configs/videogpt/dmc.yaml -------------------------------------------------------------------------------- /reds_rl/configs/videogpt/furniturebench.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/configs/videogpt/furniturebench.yaml -------------------------------------------------------------------------------- /reds_rl/configs/videogpt/furnituresim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/configs/videogpt/furnituresim.yaml -------------------------------------------------------------------------------- /reds_rl/configs/videogpt/metaworld.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/configs/videogpt/metaworld.yaml -------------------------------------------------------------------------------- /reds_rl/configs/videogpt/rlbench.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/configs/videogpt/rlbench.yaml -------------------------------------------------------------------------------- /reds_rl/configs/vqdiffusion/metaworld.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/configs/vqdiffusion/metaworld.yaml -------------------------------------------------------------------------------- /reds_rl/configs/vqgan/atari.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/configs/vqgan/atari.yaml -------------------------------------------------------------------------------- /reds_rl/configs/vqgan/dmc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/configs/vqgan/dmc.yaml -------------------------------------------------------------------------------- /reds_rl/configs/vqgan/furniturebench.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/configs/vqgan/furniturebench.yaml -------------------------------------------------------------------------------- /reds_rl/configs/vqgan/furnituresim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/configs/vqgan/furnituresim.yaml -------------------------------------------------------------------------------- /reds_rl/configs/vqgan/metaworld.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/configs/vqgan/metaworld.yaml -------------------------------------------------------------------------------- /reds_rl/configs/vqgan/rlbench.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/configs/vqgan/rlbench.yaml -------------------------------------------------------------------------------- /reds_rl/dreamerv3/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/Dockerfile -------------------------------------------------------------------------------- /reds_rl/dreamerv3/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/__init__.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/agent.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/behaviors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/behaviors.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/__init__.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/core/__init__.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/core/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/core/base.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/core/basics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/core/basics.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/core/batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/core/batch.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/core/batcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/core/batcher.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/core/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/core/checkpoint.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/core/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/core/config.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/core/counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/core/counter.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/core/distr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/core/distr.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/core/driver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/core/driver.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/core/flags.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/core/flags.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/core/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/core/logger.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/core/metaworld_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/core/metaworld_utils.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/core/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/core/metrics.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/core/parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/core/parallel.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/core/path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/core/path.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/core/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/core/random.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/core/rlbench_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/core/rlbench_utils.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/core/space.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/core/space.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/core/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/core/timer.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/core/usage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/core/usage.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/core/uuid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/core/uuid.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/core/when.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/core/when.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/core/worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/core/worker.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/core/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/core/wrappers.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/envs/atari.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/envs/atari.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/envs/cliport.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/envs/cliport.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/envs/crafter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/envs/crafter.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/envs/dmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/envs/dmc.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/envs/dmcmulticam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/envs/dmcmulticam.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/envs/dmlab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/envs/dmlab.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/envs/dummy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/envs/dummy.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/envs/factorworld.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/envs/factorworld.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/envs/from_dm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/envs/from_dm.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/envs/from_gym.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/envs/from_gym.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/envs/humanoidmulticam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/envs/humanoidmulticam.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/envs/humanoidmulticam.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/envs/humanoidmulticam.xml -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/envs/kitchen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/envs/kitchen.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/envs/loconav.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/envs/loconav.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/envs/loconav_quadruped.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/envs/loconav_quadruped.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/envs/loconav_quadruped.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/envs/loconav_quadruped.xml -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/envs/metaworld.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/envs/metaworld.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/envs/minecraft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/envs/minecraft.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/envs/minecraft_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/envs/minecraft_base.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/envs/minecraft_minerl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/envs/minecraft_minerl.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/envs/pinpad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/envs/pinpad.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/envs/rlbench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/envs/rlbench.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/envs/robodesk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/envs/robodesk.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/replay/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/replay/__init__.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/replay/chunk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/replay/chunk.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/replay/generic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/replay/generic.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/replay/limiters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/replay/limiters.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/replay/naive_chunks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/replay/naive_chunks.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/replay/replays.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/replay/replays.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/replay/reverb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/replay/reverb.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/replay/saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/replay/saver.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/replay/selectors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/replay/selectors.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/run/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/run/__init__.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/run/eval_only.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/run/eval_only.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/run/eval_only_save.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/run/eval_only_save.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/run/parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/run/parallel.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/run/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/run/train.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/run/train_amp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/run/train_amp.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/run/train_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/run/train_eval.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/run/train_holdout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/run/train_holdout.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/run/train_save.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/run/train_save.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/scripts/install-atari.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/scripts/install-atari.sh -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/scripts/install-dmlab.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/scripts/install-dmlab.sh -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/scripts/install-minecraft.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/scripts/install-minecraft.sh -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/scripts/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/scripts/plot.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/embodied/scripts/xvfb_run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/embodied/scripts/xvfb_run.sh -------------------------------------------------------------------------------- /reds_rl/dreamerv3/expl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/expl.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/jaxagent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/jaxagent.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/jaxutils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/jaxutils.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/nets.py -------------------------------------------------------------------------------- /reds_rl/dreamerv3/ninjax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/dreamerv3/ninjax.py -------------------------------------------------------------------------------- /reds_rl/videogpt/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /reds_rl/videogpt/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/videogpt/data.py -------------------------------------------------------------------------------- /reds_rl/videogpt/loss_vqgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/videogpt/loss_vqgan.py -------------------------------------------------------------------------------- /reds_rl/videogpt/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/videogpt/models/__init__.py -------------------------------------------------------------------------------- /reds_rl/videogpt/models/stylegan_disc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/videogpt/models/stylegan_disc.py -------------------------------------------------------------------------------- /reds_rl/videogpt/models/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/videogpt/models/transformer.py -------------------------------------------------------------------------------- /reds_rl/videogpt/models/videogpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/videogpt/models/videogpt.py -------------------------------------------------------------------------------- /reds_rl/videogpt/models/vqgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/videogpt/models/vqgan.py -------------------------------------------------------------------------------- /reds_rl/videogpt/reward_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/videogpt/reward_models/__init__.py -------------------------------------------------------------------------------- /reds_rl/videogpt/reward_models/videogpt_reward_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/videogpt/reward_models/videogpt_reward_model.py -------------------------------------------------------------------------------- /reds_rl/videogpt/reward_models/vqdiffusion_reward_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/videogpt/reward_models/vqdiffusion_reward_model.py -------------------------------------------------------------------------------- /reds_rl/videogpt/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/videogpt/sampler.py -------------------------------------------------------------------------------- /reds_rl/videogpt/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl/videogpt/train_utils.py -------------------------------------------------------------------------------- /reds_rl_data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl_data/__init__.py -------------------------------------------------------------------------------- /reds_rl_data/download/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl_data/download/__main__.py -------------------------------------------------------------------------------- /reds_rl_data/download/checkpoint_atari.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl_data/download/checkpoint_atari.sh -------------------------------------------------------------------------------- /reds_rl_data/download/checkpoint_dmc.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl_data/download/checkpoint_dmc.sh -------------------------------------------------------------------------------- /reds_rl_data/download/dataset_atari.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl_data/download/dataset_atari.sh -------------------------------------------------------------------------------- /reds_rl_data/download/dataset_dmc.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/reds_rl_data/download/dataset_dmc.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/collect_metaworld_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/scripts/collect_metaworld_data.py -------------------------------------------------------------------------------- /scripts/collect_rlbench_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/scripts/collect_rlbench_data.py -------------------------------------------------------------------------------- /scripts/train_dreamer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/scripts/train_dreamer.py -------------------------------------------------------------------------------- /scripts/train_reds_metaworld.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/scripts/train_reds_metaworld.sh -------------------------------------------------------------------------------- /scripts/train_reds_rlbench.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/scripts/train_reds_rlbench.sh -------------------------------------------------------------------------------- /scripts/train_videogpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/scripts/train_videogpt.py -------------------------------------------------------------------------------- /scripts/train_vqgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/scripts/train_vqgan.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csmile-1006/REDS_agent/HEAD/setup.py --------------------------------------------------------------------------------