├── .gitignore ├── LICENSE ├── README.md ├── env ├── Dockerfile ├── conda_env.yaml └── gym-0.19.0-py3-none-any.whl ├── imgs ├── mmaze.jpg ├── r2i.png └── teaser.jpg └── recall2imagine ├── Dockerfile ├── __init__.py ├── agent.py ├── behaviors.py ├── configs.yaml ├── configs_backup.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 │ ├── crafter.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 │ ├── chunk.py │ ├── generic.py │ ├── generic_lfs.py │ ├── lfs_manager.py │ ├── limiters.py │ ├── naive_chunks.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 ├── ssm ├── __init__.py ├── common.py ├── mimo.py └── siso.py ├── ssm_nets.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/README.md -------------------------------------------------------------------------------- /env/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/env/Dockerfile -------------------------------------------------------------------------------- /env/conda_env.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/env/conda_env.yaml -------------------------------------------------------------------------------- /env/gym-0.19.0-py3-none-any.whl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/env/gym-0.19.0-py3-none-any.whl -------------------------------------------------------------------------------- /imgs/mmaze.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/imgs/mmaze.jpg -------------------------------------------------------------------------------- /imgs/r2i.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/imgs/r2i.png -------------------------------------------------------------------------------- /imgs/teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/imgs/teaser.jpg -------------------------------------------------------------------------------- /recall2imagine/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/Dockerfile -------------------------------------------------------------------------------- /recall2imagine/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/__init__.py -------------------------------------------------------------------------------- /recall2imagine/agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/agent.py -------------------------------------------------------------------------------- /recall2imagine/behaviors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/behaviors.py -------------------------------------------------------------------------------- /recall2imagine/configs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/configs.yaml -------------------------------------------------------------------------------- /recall2imagine/configs_backup.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/configs_backup.yaml -------------------------------------------------------------------------------- /recall2imagine/embodied/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/__init__.py -------------------------------------------------------------------------------- /recall2imagine/embodied/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/core/__init__.py -------------------------------------------------------------------------------- /recall2imagine/embodied/core/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/core/base.py -------------------------------------------------------------------------------- /recall2imagine/embodied/core/basics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/core/basics.py -------------------------------------------------------------------------------- /recall2imagine/embodied/core/batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/core/batch.py -------------------------------------------------------------------------------- /recall2imagine/embodied/core/batcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/core/batcher.py -------------------------------------------------------------------------------- /recall2imagine/embodied/core/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/core/checkpoint.py -------------------------------------------------------------------------------- /recall2imagine/embodied/core/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/core/config.py -------------------------------------------------------------------------------- /recall2imagine/embodied/core/counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/core/counter.py -------------------------------------------------------------------------------- /recall2imagine/embodied/core/distr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/core/distr.py -------------------------------------------------------------------------------- /recall2imagine/embodied/core/driver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/core/driver.py -------------------------------------------------------------------------------- /recall2imagine/embodied/core/flags.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/core/flags.py -------------------------------------------------------------------------------- /recall2imagine/embodied/core/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/core/logger.py -------------------------------------------------------------------------------- /recall2imagine/embodied/core/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/core/metrics.py -------------------------------------------------------------------------------- /recall2imagine/embodied/core/parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/core/parallel.py -------------------------------------------------------------------------------- /recall2imagine/embodied/core/path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/core/path.py -------------------------------------------------------------------------------- /recall2imagine/embodied/core/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/core/random.py -------------------------------------------------------------------------------- /recall2imagine/embodied/core/space.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/core/space.py -------------------------------------------------------------------------------- /recall2imagine/embodied/core/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/core/timer.py -------------------------------------------------------------------------------- /recall2imagine/embodied/core/uuid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/core/uuid.py -------------------------------------------------------------------------------- /recall2imagine/embodied/core/when.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/core/when.py -------------------------------------------------------------------------------- /recall2imagine/embodied/core/worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/core/worker.py -------------------------------------------------------------------------------- /recall2imagine/embodied/core/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/core/wrappers.py -------------------------------------------------------------------------------- /recall2imagine/embodied/envs/atari.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/envs/atari.py -------------------------------------------------------------------------------- /recall2imagine/embodied/envs/crafter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/envs/crafter.py -------------------------------------------------------------------------------- /recall2imagine/embodied/envs/dmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/envs/dmc.py -------------------------------------------------------------------------------- /recall2imagine/embodied/envs/dmlab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/envs/dmlab.py -------------------------------------------------------------------------------- /recall2imagine/embodied/envs/dummy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/envs/dummy.py -------------------------------------------------------------------------------- /recall2imagine/embodied/envs/from_dm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/envs/from_dm.py -------------------------------------------------------------------------------- /recall2imagine/embodied/envs/from_gym.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/envs/from_gym.py -------------------------------------------------------------------------------- /recall2imagine/embodied/envs/loconav.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/envs/loconav.py -------------------------------------------------------------------------------- /recall2imagine/embodied/envs/loconav_quadruped.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/envs/loconav_quadruped.py -------------------------------------------------------------------------------- /recall2imagine/embodied/envs/loconav_quadruped.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/envs/loconav_quadruped.xml -------------------------------------------------------------------------------- /recall2imagine/embodied/envs/minecraft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/envs/minecraft.py -------------------------------------------------------------------------------- /recall2imagine/embodied/envs/minecraft_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/envs/minecraft_base.py -------------------------------------------------------------------------------- /recall2imagine/embodied/envs/minecraft_minerl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/envs/minecraft_minerl.py -------------------------------------------------------------------------------- /recall2imagine/embodied/envs/pinpad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/envs/pinpad.py -------------------------------------------------------------------------------- /recall2imagine/embodied/envs/robodesk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/envs/robodesk.py -------------------------------------------------------------------------------- /recall2imagine/embodied/replay/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/replay/__init__.py -------------------------------------------------------------------------------- /recall2imagine/embodied/replay/chunk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/replay/chunk.py -------------------------------------------------------------------------------- /recall2imagine/embodied/replay/generic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/replay/generic.py -------------------------------------------------------------------------------- /recall2imagine/embodied/replay/generic_lfs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/replay/generic_lfs.py -------------------------------------------------------------------------------- /recall2imagine/embodied/replay/lfs_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/replay/lfs_manager.py -------------------------------------------------------------------------------- /recall2imagine/embodied/replay/limiters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/replay/limiters.py -------------------------------------------------------------------------------- /recall2imagine/embodied/replay/naive_chunks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/replay/naive_chunks.py -------------------------------------------------------------------------------- /recall2imagine/embodied/replay/replays.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/replay/replays.py -------------------------------------------------------------------------------- /recall2imagine/embodied/replay/reverb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/replay/reverb.py -------------------------------------------------------------------------------- /recall2imagine/embodied/replay/saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/replay/saver.py -------------------------------------------------------------------------------- /recall2imagine/embodied/replay/selectors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/replay/selectors.py -------------------------------------------------------------------------------- /recall2imagine/embodied/run/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/run/__init__.py -------------------------------------------------------------------------------- /recall2imagine/embodied/run/eval_only.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/run/eval_only.py -------------------------------------------------------------------------------- /recall2imagine/embodied/run/parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/run/parallel.py -------------------------------------------------------------------------------- /recall2imagine/embodied/run/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/run/train.py -------------------------------------------------------------------------------- /recall2imagine/embodied/run/train_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/run/train_eval.py -------------------------------------------------------------------------------- /recall2imagine/embodied/run/train_holdout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/run/train_holdout.py -------------------------------------------------------------------------------- /recall2imagine/embodied/run/train_save.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/run/train_save.py -------------------------------------------------------------------------------- /recall2imagine/embodied/scripts/install-atari.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/scripts/install-atari.sh -------------------------------------------------------------------------------- /recall2imagine/embodied/scripts/install-dmlab.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/scripts/install-dmlab.sh -------------------------------------------------------------------------------- /recall2imagine/embodied/scripts/install-minecraft.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/scripts/install-minecraft.sh -------------------------------------------------------------------------------- /recall2imagine/embodied/scripts/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/scripts/plot.py -------------------------------------------------------------------------------- /recall2imagine/embodied/scripts/xvfb_run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/embodied/scripts/xvfb_run.sh -------------------------------------------------------------------------------- /recall2imagine/expl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/expl.py -------------------------------------------------------------------------------- /recall2imagine/jaxagent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/jaxagent.py -------------------------------------------------------------------------------- /recall2imagine/jaxutils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/jaxutils.py -------------------------------------------------------------------------------- /recall2imagine/nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/nets.py -------------------------------------------------------------------------------- /recall2imagine/ninjax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/ninjax.py -------------------------------------------------------------------------------- /recall2imagine/ssm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /recall2imagine/ssm/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/ssm/common.py -------------------------------------------------------------------------------- /recall2imagine/ssm/mimo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/ssm/mimo.py -------------------------------------------------------------------------------- /recall2imagine/ssm/siso.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/ssm/siso.py -------------------------------------------------------------------------------- /recall2imagine/ssm_nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/ssm_nets.py -------------------------------------------------------------------------------- /recall2imagine/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chandar-lab/Recall2Imagine/HEAD/recall2imagine/train.py --------------------------------------------------------------------------------