├── .gitignore ├── LICENSE ├── MANIFEST.in ├── README.md ├── dreamerv3 ├── Dockerfile ├── __init__.py ├── agent.py ├── behaviors.py ├── configs.yaml ├── 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 │ │ ├── metrics.py │ │ ├── parallel.py │ │ ├── path.py │ │ ├── random.py │ │ ├── space.py │ │ ├── timer.py │ │ ├── uuid.py │ │ ├── when.py │ │ ├── worker.py │ │ └── wrappers.py │ ├── envs │ │ ├── atari.py │ │ ├── cdmc.py │ │ ├── crafter.py │ │ ├── ddmc.py │ │ ├── dmc.py │ │ ├── dmlab.py │ │ ├── dummy.py │ │ ├── from_dm.py │ │ ├── from_gym.py │ │ ├── loconav.py │ │ ├── loconav_quadruped.py │ │ ├── loconav_quadruped.xml │ │ ├── minecraft.py │ │ ├── minecraft_base.py │ │ ├── minecraft_minerl.py │ │ ├── pinpad.py │ │ └── robodesk.py │ ├── replay │ │ ├── __init__.py │ │ ├── adversarial.py │ │ ├── base_prioritized_reverb.py │ │ ├── chunk.py │ │ ├── count_based.py │ │ ├── curious_replay.py │ │ ├── generic.py │ │ ├── limiters.py │ │ ├── naive_chunks.py │ │ ├── prioritized_experience_replay.py │ │ ├── replays.py │ │ ├── reverb.py │ │ ├── saver.py │ │ └── selectors.py │ ├── run │ │ ├── __init__.py │ │ ├── eval_only.py │ │ ├── parallel.py │ │ ├── train.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 ├── plot_crafter.py └── train.py ├── example.py ├── requirements.txt ├── run_local.py └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- 1 | include requirements.txt 2 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/README.md -------------------------------------------------------------------------------- /dreamerv3/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/Dockerfile -------------------------------------------------------------------------------- /dreamerv3/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/__init__.py -------------------------------------------------------------------------------- /dreamerv3/agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/agent.py -------------------------------------------------------------------------------- /dreamerv3/behaviors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/behaviors.py -------------------------------------------------------------------------------- /dreamerv3/configs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/configs.yaml -------------------------------------------------------------------------------- /dreamerv3/embodied/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/__init__.py -------------------------------------------------------------------------------- /dreamerv3/embodied/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/core/__init__.py -------------------------------------------------------------------------------- /dreamerv3/embodied/core/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/core/base.py -------------------------------------------------------------------------------- /dreamerv3/embodied/core/basics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/core/basics.py -------------------------------------------------------------------------------- /dreamerv3/embodied/core/batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/core/batch.py -------------------------------------------------------------------------------- /dreamerv3/embodied/core/batcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/core/batcher.py -------------------------------------------------------------------------------- /dreamerv3/embodied/core/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/core/checkpoint.py -------------------------------------------------------------------------------- /dreamerv3/embodied/core/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/core/config.py -------------------------------------------------------------------------------- /dreamerv3/embodied/core/counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/core/counter.py -------------------------------------------------------------------------------- /dreamerv3/embodied/core/distr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/core/distr.py -------------------------------------------------------------------------------- /dreamerv3/embodied/core/driver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/core/driver.py -------------------------------------------------------------------------------- /dreamerv3/embodied/core/flags.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/core/flags.py -------------------------------------------------------------------------------- /dreamerv3/embodied/core/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/core/logger.py -------------------------------------------------------------------------------- /dreamerv3/embodied/core/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/core/metrics.py -------------------------------------------------------------------------------- /dreamerv3/embodied/core/parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/core/parallel.py -------------------------------------------------------------------------------- /dreamerv3/embodied/core/path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/core/path.py -------------------------------------------------------------------------------- /dreamerv3/embodied/core/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/core/random.py -------------------------------------------------------------------------------- /dreamerv3/embodied/core/space.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/core/space.py -------------------------------------------------------------------------------- /dreamerv3/embodied/core/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/core/timer.py -------------------------------------------------------------------------------- /dreamerv3/embodied/core/uuid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/core/uuid.py -------------------------------------------------------------------------------- /dreamerv3/embodied/core/when.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/core/when.py -------------------------------------------------------------------------------- /dreamerv3/embodied/core/worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/core/worker.py -------------------------------------------------------------------------------- /dreamerv3/embodied/core/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/core/wrappers.py -------------------------------------------------------------------------------- /dreamerv3/embodied/envs/atari.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/envs/atari.py -------------------------------------------------------------------------------- /dreamerv3/embodied/envs/cdmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/envs/cdmc.py -------------------------------------------------------------------------------- /dreamerv3/embodied/envs/crafter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/envs/crafter.py -------------------------------------------------------------------------------- /dreamerv3/embodied/envs/ddmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/envs/ddmc.py -------------------------------------------------------------------------------- /dreamerv3/embodied/envs/dmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/envs/dmc.py -------------------------------------------------------------------------------- /dreamerv3/embodied/envs/dmlab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/envs/dmlab.py -------------------------------------------------------------------------------- /dreamerv3/embodied/envs/dummy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/envs/dummy.py -------------------------------------------------------------------------------- /dreamerv3/embodied/envs/from_dm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/envs/from_dm.py -------------------------------------------------------------------------------- /dreamerv3/embodied/envs/from_gym.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/envs/from_gym.py -------------------------------------------------------------------------------- /dreamerv3/embodied/envs/loconav.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/envs/loconav.py -------------------------------------------------------------------------------- /dreamerv3/embodied/envs/loconav_quadruped.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/envs/loconav_quadruped.py -------------------------------------------------------------------------------- /dreamerv3/embodied/envs/loconav_quadruped.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/envs/loconav_quadruped.xml -------------------------------------------------------------------------------- /dreamerv3/embodied/envs/minecraft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/envs/minecraft.py -------------------------------------------------------------------------------- /dreamerv3/embodied/envs/minecraft_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/envs/minecraft_base.py -------------------------------------------------------------------------------- /dreamerv3/embodied/envs/minecraft_minerl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/envs/minecraft_minerl.py -------------------------------------------------------------------------------- /dreamerv3/embodied/envs/pinpad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/envs/pinpad.py -------------------------------------------------------------------------------- /dreamerv3/embodied/envs/robodesk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/envs/robodesk.py -------------------------------------------------------------------------------- /dreamerv3/embodied/replay/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/replay/__init__.py -------------------------------------------------------------------------------- /dreamerv3/embodied/replay/adversarial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/replay/adversarial.py -------------------------------------------------------------------------------- /dreamerv3/embodied/replay/base_prioritized_reverb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/replay/base_prioritized_reverb.py -------------------------------------------------------------------------------- /dreamerv3/embodied/replay/chunk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/replay/chunk.py -------------------------------------------------------------------------------- /dreamerv3/embodied/replay/count_based.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/replay/count_based.py -------------------------------------------------------------------------------- /dreamerv3/embodied/replay/curious_replay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/replay/curious_replay.py -------------------------------------------------------------------------------- /dreamerv3/embodied/replay/generic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/replay/generic.py -------------------------------------------------------------------------------- /dreamerv3/embodied/replay/limiters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/replay/limiters.py -------------------------------------------------------------------------------- /dreamerv3/embodied/replay/naive_chunks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/replay/naive_chunks.py -------------------------------------------------------------------------------- /dreamerv3/embodied/replay/prioritized_experience_replay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/replay/prioritized_experience_replay.py -------------------------------------------------------------------------------- /dreamerv3/embodied/replay/replays.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/replay/replays.py -------------------------------------------------------------------------------- /dreamerv3/embodied/replay/reverb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/replay/reverb.py -------------------------------------------------------------------------------- /dreamerv3/embodied/replay/saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/replay/saver.py -------------------------------------------------------------------------------- /dreamerv3/embodied/replay/selectors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/replay/selectors.py -------------------------------------------------------------------------------- /dreamerv3/embodied/run/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/run/__init__.py -------------------------------------------------------------------------------- /dreamerv3/embodied/run/eval_only.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/run/eval_only.py -------------------------------------------------------------------------------- /dreamerv3/embodied/run/parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/run/parallel.py -------------------------------------------------------------------------------- /dreamerv3/embodied/run/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/run/train.py -------------------------------------------------------------------------------- /dreamerv3/embodied/run/train_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/run/train_eval.py -------------------------------------------------------------------------------- /dreamerv3/embodied/run/train_holdout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/run/train_holdout.py -------------------------------------------------------------------------------- /dreamerv3/embodied/run/train_save.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/run/train_save.py -------------------------------------------------------------------------------- /dreamerv3/embodied/scripts/install-atari.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/scripts/install-atari.sh -------------------------------------------------------------------------------- /dreamerv3/embodied/scripts/install-dmlab.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/scripts/install-dmlab.sh -------------------------------------------------------------------------------- /dreamerv3/embodied/scripts/install-minecraft.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/scripts/install-minecraft.sh -------------------------------------------------------------------------------- /dreamerv3/embodied/scripts/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/scripts/plot.py -------------------------------------------------------------------------------- /dreamerv3/embodied/scripts/xvfb_run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/embodied/scripts/xvfb_run.sh -------------------------------------------------------------------------------- /dreamerv3/expl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/expl.py -------------------------------------------------------------------------------- /dreamerv3/jaxagent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/jaxagent.py -------------------------------------------------------------------------------- /dreamerv3/jaxutils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/jaxutils.py -------------------------------------------------------------------------------- /dreamerv3/nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/nets.py -------------------------------------------------------------------------------- /dreamerv3/ninjax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/ninjax.py -------------------------------------------------------------------------------- /dreamerv3/plot_crafter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/plot_crafter.py -------------------------------------------------------------------------------- /dreamerv3/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/dreamerv3/train.py -------------------------------------------------------------------------------- /example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/example.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_local.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/run_local.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AutonomousAgentsLab/cr-dv3/HEAD/setup.py --------------------------------------------------------------------------------