├── .gitignore ├── LICENSE ├── README.md ├── experiments ├── antmaze │ └── train_icvf.py └── xmagical │ └── train_icvf.py ├── icvf_envs ├── antmaze │ ├── ant_diagnostics.py │ ├── antmaze_aux │ │ ├── antmaze-large-diverse-v2-aux.npz │ │ ├── antmaze-large-play-v2-aux.npz │ │ ├── antmaze-medium-diverse-v2-aux.npz │ │ └── antmaze-medium-play-v2-aux.npz │ ├── d4rl_ant.py │ ├── d4rl_pm.py │ └── d4rl_utils.py ├── atari │ ├── __init__.py │ ├── atari_envs.py │ └── offline_env.py ├── maze_nav.py └── xmagical │ ├── __init__.py │ └── utils.py ├── jaxrl_m ├── README.md ├── common.py ├── dataset.py ├── evaluation.py ├── networks.py ├── typing.py ├── vision │ ├── __init__.py │ ├── bigvision_common.py │ ├── bigvision_resnetv2.py │ ├── bigvision_utils.py │ ├── data_augmentations.py │ ├── impala.py │ ├── resnet_v1.py │ └── small_encoders.py └── wandb.py ├── requirements.txt └── src ├── __init__.py ├── gc_dataset.py ├── icvf_learner.py ├── icvf_networks.py └── viz_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/README.md -------------------------------------------------------------------------------- /experiments/antmaze/train_icvf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/experiments/antmaze/train_icvf.py -------------------------------------------------------------------------------- /experiments/xmagical/train_icvf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/experiments/xmagical/train_icvf.py -------------------------------------------------------------------------------- /icvf_envs/antmaze/ant_diagnostics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/icvf_envs/antmaze/ant_diagnostics.py -------------------------------------------------------------------------------- /icvf_envs/antmaze/antmaze_aux/antmaze-large-diverse-v2-aux.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/icvf_envs/antmaze/antmaze_aux/antmaze-large-diverse-v2-aux.npz -------------------------------------------------------------------------------- /icvf_envs/antmaze/antmaze_aux/antmaze-large-play-v2-aux.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/icvf_envs/antmaze/antmaze_aux/antmaze-large-play-v2-aux.npz -------------------------------------------------------------------------------- /icvf_envs/antmaze/antmaze_aux/antmaze-medium-diverse-v2-aux.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/icvf_envs/antmaze/antmaze_aux/antmaze-medium-diverse-v2-aux.npz -------------------------------------------------------------------------------- /icvf_envs/antmaze/antmaze_aux/antmaze-medium-play-v2-aux.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/icvf_envs/antmaze/antmaze_aux/antmaze-medium-play-v2-aux.npz -------------------------------------------------------------------------------- /icvf_envs/antmaze/d4rl_ant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/icvf_envs/antmaze/d4rl_ant.py -------------------------------------------------------------------------------- /icvf_envs/antmaze/d4rl_pm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/icvf_envs/antmaze/d4rl_pm.py -------------------------------------------------------------------------------- /icvf_envs/antmaze/d4rl_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/icvf_envs/antmaze/d4rl_utils.py -------------------------------------------------------------------------------- /icvf_envs/atari/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/icvf_envs/atari/__init__.py -------------------------------------------------------------------------------- /icvf_envs/atari/atari_envs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/icvf_envs/atari/atari_envs.py -------------------------------------------------------------------------------- /icvf_envs/atari/offline_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/icvf_envs/atari/offline_env.py -------------------------------------------------------------------------------- /icvf_envs/maze_nav.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/icvf_envs/maze_nav.py -------------------------------------------------------------------------------- /icvf_envs/xmagical/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/icvf_envs/xmagical/__init__.py -------------------------------------------------------------------------------- /icvf_envs/xmagical/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/icvf_envs/xmagical/utils.py -------------------------------------------------------------------------------- /jaxrl_m/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/jaxrl_m/README.md -------------------------------------------------------------------------------- /jaxrl_m/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/jaxrl_m/common.py -------------------------------------------------------------------------------- /jaxrl_m/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/jaxrl_m/dataset.py -------------------------------------------------------------------------------- /jaxrl_m/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/jaxrl_m/evaluation.py -------------------------------------------------------------------------------- /jaxrl_m/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/jaxrl_m/networks.py -------------------------------------------------------------------------------- /jaxrl_m/typing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/jaxrl_m/typing.py -------------------------------------------------------------------------------- /jaxrl_m/vision/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/jaxrl_m/vision/__init__.py -------------------------------------------------------------------------------- /jaxrl_m/vision/bigvision_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/jaxrl_m/vision/bigvision_common.py -------------------------------------------------------------------------------- /jaxrl_m/vision/bigvision_resnetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/jaxrl_m/vision/bigvision_resnetv2.py -------------------------------------------------------------------------------- /jaxrl_m/vision/bigvision_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/jaxrl_m/vision/bigvision_utils.py -------------------------------------------------------------------------------- /jaxrl_m/vision/data_augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/jaxrl_m/vision/data_augmentations.py -------------------------------------------------------------------------------- /jaxrl_m/vision/impala.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/jaxrl_m/vision/impala.py -------------------------------------------------------------------------------- /jaxrl_m/vision/resnet_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/jaxrl_m/vision/resnet_v1.py -------------------------------------------------------------------------------- /jaxrl_m/vision/small_encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/jaxrl_m/vision/small_encoders.py -------------------------------------------------------------------------------- /jaxrl_m/wandb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/jaxrl_m/wandb.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/gc_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/src/gc_dataset.py -------------------------------------------------------------------------------- /src/icvf_learner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/src/icvf_learner.py -------------------------------------------------------------------------------- /src/icvf_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/src/icvf_networks.py -------------------------------------------------------------------------------- /src/viz_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dibyaghosh/icvf_release/HEAD/src/viz_utils.py --------------------------------------------------------------------------------