├── .gitignore ├── Dockerfile ├── LICENSE ├── README.md ├── contributing.md ├── data_processing ├── bridgedata_numpy_to_tfrecord.py └── bridgedata_raw_to_numpy.py ├── experiments ├── configs │ ├── data_config.py │ └── train_config.py ├── eval.py ├── eval_gc.py ├── eval_lc.py ├── susie │ └── calvin │ │ ├── README.md │ │ ├── calvin_gcbc.py │ │ ├── calvin_lcbc.py │ │ ├── configs │ │ ├── gcbc_data_config.py │ │ ├── gcbc_train_config.py │ │ ├── lcbc_data_config.py │ │ └── lcbc_train_config.py │ │ ├── dataset_conversion_scripts │ │ ├── goal_conditioned.py │ │ └── language_conditioned.py │ │ └── scripts │ │ ├── launch_calvin_gcbc.sh │ │ └── launch_calvin_lcbc.sh ├── train.py └── utils.py ├── jaxrl_m ├── agents │ ├── __init__.py │ └── continuous │ │ ├── bc.py │ │ ├── gc_bc.py │ │ ├── gc_ddpm_bc.py │ │ ├── gc_iql.py │ │ ├── iql.py │ │ ├── lc_bc.py │ │ └── stable_contrastive_rl.py ├── common │ ├── common.py │ ├── encoding.py │ ├── typing.py │ └── wandb.py ├── data │ ├── bridge_dataset.py │ ├── calvin_dataset.py │ ├── text_processing.py │ ├── tf_augmentations.py │ └── tf_goal_relabeling.py ├── networks │ ├── actor_critic_nets.py │ ├── diffusion_nets.py │ └── mlp.py ├── utils │ └── timer_utils.py └── vision │ ├── __init__.py │ ├── film_conditioning_layer.py │ └── resnet_v1.py ├── requirements.txt └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/.gitignore -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/Dockerfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/README.md -------------------------------------------------------------------------------- /contributing.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/contributing.md -------------------------------------------------------------------------------- /data_processing/bridgedata_numpy_to_tfrecord.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/data_processing/bridgedata_numpy_to_tfrecord.py -------------------------------------------------------------------------------- /data_processing/bridgedata_raw_to_numpy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/data_processing/bridgedata_raw_to_numpy.py -------------------------------------------------------------------------------- /experiments/configs/data_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/experiments/configs/data_config.py -------------------------------------------------------------------------------- /experiments/configs/train_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/experiments/configs/train_config.py -------------------------------------------------------------------------------- /experiments/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/experiments/eval.py -------------------------------------------------------------------------------- /experiments/eval_gc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/experiments/eval_gc.py -------------------------------------------------------------------------------- /experiments/eval_lc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/experiments/eval_lc.py -------------------------------------------------------------------------------- /experiments/susie/calvin/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/experiments/susie/calvin/README.md -------------------------------------------------------------------------------- /experiments/susie/calvin/calvin_gcbc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/experiments/susie/calvin/calvin_gcbc.py -------------------------------------------------------------------------------- /experiments/susie/calvin/calvin_lcbc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/experiments/susie/calvin/calvin_lcbc.py -------------------------------------------------------------------------------- /experiments/susie/calvin/configs/gcbc_data_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/experiments/susie/calvin/configs/gcbc_data_config.py -------------------------------------------------------------------------------- /experiments/susie/calvin/configs/gcbc_train_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/experiments/susie/calvin/configs/gcbc_train_config.py -------------------------------------------------------------------------------- /experiments/susie/calvin/configs/lcbc_data_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/experiments/susie/calvin/configs/lcbc_data_config.py -------------------------------------------------------------------------------- /experiments/susie/calvin/configs/lcbc_train_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/experiments/susie/calvin/configs/lcbc_train_config.py -------------------------------------------------------------------------------- /experiments/susie/calvin/dataset_conversion_scripts/goal_conditioned.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/experiments/susie/calvin/dataset_conversion_scripts/goal_conditioned.py -------------------------------------------------------------------------------- /experiments/susie/calvin/dataset_conversion_scripts/language_conditioned.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/experiments/susie/calvin/dataset_conversion_scripts/language_conditioned.py -------------------------------------------------------------------------------- /experiments/susie/calvin/scripts/launch_calvin_gcbc.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/experiments/susie/calvin/scripts/launch_calvin_gcbc.sh -------------------------------------------------------------------------------- /experiments/susie/calvin/scripts/launch_calvin_lcbc.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/experiments/susie/calvin/scripts/launch_calvin_lcbc.sh -------------------------------------------------------------------------------- /experiments/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/experiments/train.py -------------------------------------------------------------------------------- /experiments/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/experiments/utils.py -------------------------------------------------------------------------------- /jaxrl_m/agents/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/jaxrl_m/agents/__init__.py -------------------------------------------------------------------------------- /jaxrl_m/agents/continuous/bc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/jaxrl_m/agents/continuous/bc.py -------------------------------------------------------------------------------- /jaxrl_m/agents/continuous/gc_bc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/jaxrl_m/agents/continuous/gc_bc.py -------------------------------------------------------------------------------- /jaxrl_m/agents/continuous/gc_ddpm_bc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/jaxrl_m/agents/continuous/gc_ddpm_bc.py -------------------------------------------------------------------------------- /jaxrl_m/agents/continuous/gc_iql.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/jaxrl_m/agents/continuous/gc_iql.py -------------------------------------------------------------------------------- /jaxrl_m/agents/continuous/iql.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/jaxrl_m/agents/continuous/iql.py -------------------------------------------------------------------------------- /jaxrl_m/agents/continuous/lc_bc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/jaxrl_m/agents/continuous/lc_bc.py -------------------------------------------------------------------------------- /jaxrl_m/agents/continuous/stable_contrastive_rl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/jaxrl_m/agents/continuous/stable_contrastive_rl.py -------------------------------------------------------------------------------- /jaxrl_m/common/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/jaxrl_m/common/common.py -------------------------------------------------------------------------------- /jaxrl_m/common/encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/jaxrl_m/common/encoding.py -------------------------------------------------------------------------------- /jaxrl_m/common/typing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/jaxrl_m/common/typing.py -------------------------------------------------------------------------------- /jaxrl_m/common/wandb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/jaxrl_m/common/wandb.py -------------------------------------------------------------------------------- /jaxrl_m/data/bridge_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/jaxrl_m/data/bridge_dataset.py -------------------------------------------------------------------------------- /jaxrl_m/data/calvin_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/jaxrl_m/data/calvin_dataset.py -------------------------------------------------------------------------------- /jaxrl_m/data/text_processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/jaxrl_m/data/text_processing.py -------------------------------------------------------------------------------- /jaxrl_m/data/tf_augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/jaxrl_m/data/tf_augmentations.py -------------------------------------------------------------------------------- /jaxrl_m/data/tf_goal_relabeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/jaxrl_m/data/tf_goal_relabeling.py -------------------------------------------------------------------------------- /jaxrl_m/networks/actor_critic_nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/jaxrl_m/networks/actor_critic_nets.py -------------------------------------------------------------------------------- /jaxrl_m/networks/diffusion_nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/jaxrl_m/networks/diffusion_nets.py -------------------------------------------------------------------------------- /jaxrl_m/networks/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/jaxrl_m/networks/mlp.py -------------------------------------------------------------------------------- /jaxrl_m/utils/timer_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/jaxrl_m/utils/timer_utils.py -------------------------------------------------------------------------------- /jaxrl_m/vision/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/jaxrl_m/vision/__init__.py -------------------------------------------------------------------------------- /jaxrl_m/vision/film_conditioning_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/jaxrl_m/vision/film_conditioning_layer.py -------------------------------------------------------------------------------- /jaxrl_m/vision/resnet_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/jaxrl_m/vision/resnet_v1.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rail-berkeley/bridge_data_v2/HEAD/setup.py --------------------------------------------------------------------------------