├── .flake8 ├── .gitignore ├── .gitmodules ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── config ├── calvin │ ├── eval │ │ └── close_drawer │ │ │ ├── eval_diffusion_mlp_feat_vision.yaml │ │ │ ├── eval_diffusion_mlp_img.yaml │ │ │ ├── eval_ft_diffusion_mlp_feat_vision.yaml │ │ │ └── eval_ft_diffusion_mlp_img.yaml │ ├── finetune │ │ ├── close_drawer │ │ │ ├── ft_mb_ppo_diffusion_mlp_feat_vision.yaml │ │ │ ├── ft_ppo_diffusion_mlp_feat_vision.yaml │ │ │ ├── ft_ppo_diffusion_mlp_img.yaml │ │ │ └── ft_ppo_diffusion_mlp_state.yaml │ │ └── turn_on_led │ │ │ └── ft_mb_ppo_diffusion_mlp_feat_vision.yaml │ └── pretrain │ │ ├── close_drawer │ │ ├── pre_diffusion_mlp_feat_vision.yaml │ │ ├── pre_diffusion_mlp_img.yaml │ │ └── pre_diffusion_mlp_state.yaml │ │ ├── move_slider_left │ │ ├── pre_diffusion_mlp_feat_vision.yaml │ │ ├── pre_diffusion_mlp_img.yaml │ │ └── pre_diffusion_mlp_state.yaml │ │ ├── move_slider_right │ │ ├── pre_diffusion_mlp_feat_vision.yaml │ │ ├── pre_diffusion_mlp_img.yaml │ │ └── pre_diffusion_mlp_state.yaml │ │ ├── open_drawer │ │ ├── pre_diffusion_mlp_feat_vision.yaml │ │ ├── pre_diffusion_mlp_img.yaml │ │ └── pre_diffusion_mlp_state.yaml │ │ ├── turn_off_led │ │ ├── pre_diffusion_mlp_feat_vision.yaml │ │ ├── pre_diffusion_mlp_img.yaml │ │ └── pre_diffusion_mlp_state.yaml │ │ ├── turn_off_lightbulb │ │ ├── pre_diffusion_mlp_feat_vision.yaml │ │ ├── pre_diffusion_mlp_img.yaml │ │ └── pre_diffusion_mlp_state.yaml │ │ ├── turn_on_led │ │ ├── pre_diffusion_mlp_feat_vision.yaml │ │ ├── pre_diffusion_mlp_img.yaml │ │ └── pre_diffusion_mlp_state.yaml │ │ └── turn_on_lightbulb │ │ ├── pre_diffusion_mlp_feat_vision.yaml │ │ ├── pre_diffusion_mlp_img.yaml │ │ └── pre_diffusion_mlp_state.yaml ├── dataset │ ├── calculate_stats.yaml │ ├── extract_expert_demos_feat.yaml │ ├── extract_expert_demos_feat_libero.yaml │ ├── extract_expert_demos_img.yaml │ ├── extract_expert_demos_libero.yaml │ ├── extract_expert_demos_state.yaml │ ├── from_libero_to_calvin.yaml │ ├── generate_init_obs_libero.yaml │ ├── preprocess_calvin.yaml │ ├── preprocess_calvin_state.yaml │ ├── preprocess_libero.yaml │ ├── regenerate_libero.yaml │ └── split_calvin.yaml ├── env │ └── calvin.yaml ├── featurizer.yaml ├── featurizer_libero.yaml ├── libero │ ├── finetune │ │ └── kitchen_scene1 │ │ │ └── ft_mb_ppo_diffusion_mlp_feat_vision.yaml │ └── pretrain │ │ └── kitchen_scene1 │ │ ├── pre_diffusion_mlp_feat_vision.yaml │ │ ├── pre_diffusion_mlp_img.yaml │ │ └── pre_diffusion_mlp_state.yaml ├── logger │ ├── none.yaml │ └── wandb.yaml ├── rewcls │ ├── balance_rewcls_dataset.yaml │ ├── balance_rewcls_dataset_libero.yaml │ ├── contrastive.yaml │ ├── contrastive_img.yaml │ ├── contrastive_libero.yaml │ ├── generate_rewards.yaml │ ├── generate_rewards_libero.yaml │ ├── rewcls_dataset.yaml │ ├── rewcls_dataset_libero.yaml │ └── rewcls_img_dataset.yaml ├── setup.yaml ├── train_hybrid_wm.yaml ├── train_state_wm.yaml ├── train_wm.yaml └── train_wm_libero.yaml ├── dataset ├── README.md └── download_data.sh ├── diwa ├── __init__.py ├── agent │ ├── eval │ │ ├── eval_agent.py │ │ ├── eval_diffusion_agent.py │ │ └── eval_diffusion_img_agent.py │ ├── finetune │ │ ├── train_agent.py │ │ ├── train_mb_ppo_diffusion_agent_visionwm.py │ │ ├── train_ppo_agent.py │ │ ├── train_ppo_diffusion_agent.py │ │ └── train_ppo_diffusion_img_agent.py │ └── pretrain │ │ ├── train_agent.py │ │ ├── train_diffusion_agent.py │ │ └── train_gaussian_agent.py ├── dataset │ ├── __init__.py │ ├── rewcls_dataset.py │ └── sequence.py ├── env │ ├── runner │ │ ├── base_runner.py │ │ ├── hybridwm.py │ │ ├── hybridwm_gaussian.py │ │ ├── image.py │ │ ├── statewm.py │ │ ├── visionwm.py │ │ └── visionwm_gaussian.py │ ├── utils │ │ ├── __init__.py │ │ ├── calvin_helpers.py │ │ ├── custom_subprocvec.py │ │ └── custom_subprocvec_libero.py │ └── wrapper │ │ ├── __init__.py │ │ ├── base_calvin.py │ │ ├── base_libero.py │ │ ├── calvin_image.py │ │ ├── calvin_image_stacked.py │ │ ├── calvin_image_stateful.py │ │ ├── calvin_lowdim.py │ │ ├── libero_image.py │ │ ├── libero_image_stacked.py │ │ ├── libero_image_stateful.py │ │ ├── libero_lowdim.py │ │ └── multistep.py ├── model │ ├── common │ │ ├── __init__.py │ │ ├── critic.py │ │ ├── gaussian.py │ │ ├── mlp.py │ │ ├── mlp_gaussian.py │ │ ├── modules.py │ │ ├── transformer.py │ │ └── vit.py │ ├── diffusion │ │ ├── __init__.py │ │ ├── diffusion.py │ │ ├── diffusion_eval.py │ │ ├── diffusion_ppo.py │ │ ├── diffusion_vpg.py │ │ ├── eta.py │ │ ├── mlp_diffusion.py │ │ ├── modules.py │ │ ├── sampling.py │ │ ├── sde_lib.py │ │ └── unet.py │ ├── gaussian │ │ ├── __init__.py │ │ ├── gaussian_ppo.py │ │ └── gaussian_vpg.py │ └── rewcls │ │ ├── contrastive.py │ │ └── resnet_contrastive.py ├── utils │ ├── config_fix.py │ ├── pytorch3d_transforms.py │ ├── reward_scaling.py │ ├── rotation_transformer.py │ ├── scheduler.py │ ├── timer.py │ └── transforms.py └── wm │ ├── encoder │ ├── base.py │ ├── hybridwm.py │ ├── statewm.py │ └── visionwm.py │ ├── utils.py │ └── wrapper │ ├── base.py │ ├── hybridwm.py │ ├── hybridwm_cls.py │ ├── statewm.py │ └── visionwm_cls.py ├── docs ├── diwa_cover.png └── diwa_overview.png ├── install.sh ├── pyproject.toml ├── requirements-dev.txt ├── requirements-libero.txt ├── requirements.txt ├── scripts ├── dataset │ ├── calculate_meanstd_stats.py │ ├── calculate_minmax_stats.py │ ├── convert_libero_to_calvin.py │ ├── extract_expert_demos_feat.py │ ├── extract_expert_demos_img.py │ ├── extract_expert_demos_state.py │ ├── generate_init_obs_libero.py │ ├── preprocess_calvin_dataset.py │ ├── regenerate_libero_dataset.py │ └── split_calvin_dataset.py ├── featurizer.py ├── rewcls │ ├── balance_rewcls_dataset.py │ ├── balance_rewcls_img_dataset.py │ ├── create_rewcls_dataset.py │ ├── create_rewcls_img_dataset.py │ ├── generate_rewards_calvin.py │ ├── generate_rewards_libero.py │ ├── train_contrastive.py │ └── train_img_contrastive.py ├── run.py ├── set_path.sh ├── tests │ └── visionwm.py └── train_wm.py └── setup.py /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/.flake8 -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/.gitmodules -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/README.md -------------------------------------------------------------------------------- /config/calvin/eval/close_drawer/eval_diffusion_mlp_feat_vision.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/eval/close_drawer/eval_diffusion_mlp_feat_vision.yaml -------------------------------------------------------------------------------- /config/calvin/eval/close_drawer/eval_diffusion_mlp_img.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/eval/close_drawer/eval_diffusion_mlp_img.yaml -------------------------------------------------------------------------------- /config/calvin/eval/close_drawer/eval_ft_diffusion_mlp_feat_vision.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/eval/close_drawer/eval_ft_diffusion_mlp_feat_vision.yaml -------------------------------------------------------------------------------- /config/calvin/eval/close_drawer/eval_ft_diffusion_mlp_img.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/eval/close_drawer/eval_ft_diffusion_mlp_img.yaml -------------------------------------------------------------------------------- /config/calvin/finetune/close_drawer/ft_mb_ppo_diffusion_mlp_feat_vision.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/finetune/close_drawer/ft_mb_ppo_diffusion_mlp_feat_vision.yaml -------------------------------------------------------------------------------- /config/calvin/finetune/close_drawer/ft_ppo_diffusion_mlp_feat_vision.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/finetune/close_drawer/ft_ppo_diffusion_mlp_feat_vision.yaml -------------------------------------------------------------------------------- /config/calvin/finetune/close_drawer/ft_ppo_diffusion_mlp_img.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/finetune/close_drawer/ft_ppo_diffusion_mlp_img.yaml -------------------------------------------------------------------------------- /config/calvin/finetune/close_drawer/ft_ppo_diffusion_mlp_state.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/finetune/close_drawer/ft_ppo_diffusion_mlp_state.yaml -------------------------------------------------------------------------------- /config/calvin/finetune/turn_on_led/ft_mb_ppo_diffusion_mlp_feat_vision.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/finetune/turn_on_led/ft_mb_ppo_diffusion_mlp_feat_vision.yaml -------------------------------------------------------------------------------- /config/calvin/pretrain/close_drawer/pre_diffusion_mlp_feat_vision.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/pretrain/close_drawer/pre_diffusion_mlp_feat_vision.yaml -------------------------------------------------------------------------------- /config/calvin/pretrain/close_drawer/pre_diffusion_mlp_img.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/pretrain/close_drawer/pre_diffusion_mlp_img.yaml -------------------------------------------------------------------------------- /config/calvin/pretrain/close_drawer/pre_diffusion_mlp_state.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/pretrain/close_drawer/pre_diffusion_mlp_state.yaml -------------------------------------------------------------------------------- /config/calvin/pretrain/move_slider_left/pre_diffusion_mlp_feat_vision.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/pretrain/move_slider_left/pre_diffusion_mlp_feat_vision.yaml -------------------------------------------------------------------------------- /config/calvin/pretrain/move_slider_left/pre_diffusion_mlp_img.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/pretrain/move_slider_left/pre_diffusion_mlp_img.yaml -------------------------------------------------------------------------------- /config/calvin/pretrain/move_slider_left/pre_diffusion_mlp_state.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/pretrain/move_slider_left/pre_diffusion_mlp_state.yaml -------------------------------------------------------------------------------- /config/calvin/pretrain/move_slider_right/pre_diffusion_mlp_feat_vision.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/pretrain/move_slider_right/pre_diffusion_mlp_feat_vision.yaml -------------------------------------------------------------------------------- /config/calvin/pretrain/move_slider_right/pre_diffusion_mlp_img.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/pretrain/move_slider_right/pre_diffusion_mlp_img.yaml -------------------------------------------------------------------------------- /config/calvin/pretrain/move_slider_right/pre_diffusion_mlp_state.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/pretrain/move_slider_right/pre_diffusion_mlp_state.yaml -------------------------------------------------------------------------------- /config/calvin/pretrain/open_drawer/pre_diffusion_mlp_feat_vision.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/pretrain/open_drawer/pre_diffusion_mlp_feat_vision.yaml -------------------------------------------------------------------------------- /config/calvin/pretrain/open_drawer/pre_diffusion_mlp_img.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/pretrain/open_drawer/pre_diffusion_mlp_img.yaml -------------------------------------------------------------------------------- /config/calvin/pretrain/open_drawer/pre_diffusion_mlp_state.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/pretrain/open_drawer/pre_diffusion_mlp_state.yaml -------------------------------------------------------------------------------- /config/calvin/pretrain/turn_off_led/pre_diffusion_mlp_feat_vision.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/pretrain/turn_off_led/pre_diffusion_mlp_feat_vision.yaml -------------------------------------------------------------------------------- /config/calvin/pretrain/turn_off_led/pre_diffusion_mlp_img.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/pretrain/turn_off_led/pre_diffusion_mlp_img.yaml -------------------------------------------------------------------------------- /config/calvin/pretrain/turn_off_led/pre_diffusion_mlp_state.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/pretrain/turn_off_led/pre_diffusion_mlp_state.yaml -------------------------------------------------------------------------------- /config/calvin/pretrain/turn_off_lightbulb/pre_diffusion_mlp_feat_vision.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/pretrain/turn_off_lightbulb/pre_diffusion_mlp_feat_vision.yaml -------------------------------------------------------------------------------- /config/calvin/pretrain/turn_off_lightbulb/pre_diffusion_mlp_img.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/pretrain/turn_off_lightbulb/pre_diffusion_mlp_img.yaml -------------------------------------------------------------------------------- /config/calvin/pretrain/turn_off_lightbulb/pre_diffusion_mlp_state.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/pretrain/turn_off_lightbulb/pre_diffusion_mlp_state.yaml -------------------------------------------------------------------------------- /config/calvin/pretrain/turn_on_led/pre_diffusion_mlp_feat_vision.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/pretrain/turn_on_led/pre_diffusion_mlp_feat_vision.yaml -------------------------------------------------------------------------------- /config/calvin/pretrain/turn_on_led/pre_diffusion_mlp_img.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/pretrain/turn_on_led/pre_diffusion_mlp_img.yaml -------------------------------------------------------------------------------- /config/calvin/pretrain/turn_on_led/pre_diffusion_mlp_state.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/pretrain/turn_on_led/pre_diffusion_mlp_state.yaml -------------------------------------------------------------------------------- /config/calvin/pretrain/turn_on_lightbulb/pre_diffusion_mlp_feat_vision.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/pretrain/turn_on_lightbulb/pre_diffusion_mlp_feat_vision.yaml -------------------------------------------------------------------------------- /config/calvin/pretrain/turn_on_lightbulb/pre_diffusion_mlp_img.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/pretrain/turn_on_lightbulb/pre_diffusion_mlp_img.yaml -------------------------------------------------------------------------------- /config/calvin/pretrain/turn_on_lightbulb/pre_diffusion_mlp_state.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/calvin/pretrain/turn_on_lightbulb/pre_diffusion_mlp_state.yaml -------------------------------------------------------------------------------- /config/dataset/calculate_stats.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/dataset/calculate_stats.yaml -------------------------------------------------------------------------------- /config/dataset/extract_expert_demos_feat.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/dataset/extract_expert_demos_feat.yaml -------------------------------------------------------------------------------- /config/dataset/extract_expert_demos_feat_libero.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/dataset/extract_expert_demos_feat_libero.yaml -------------------------------------------------------------------------------- /config/dataset/extract_expert_demos_img.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/dataset/extract_expert_demos_img.yaml -------------------------------------------------------------------------------- /config/dataset/extract_expert_demos_libero.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/dataset/extract_expert_demos_libero.yaml -------------------------------------------------------------------------------- /config/dataset/extract_expert_demos_state.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/dataset/extract_expert_demos_state.yaml -------------------------------------------------------------------------------- /config/dataset/from_libero_to_calvin.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/dataset/from_libero_to_calvin.yaml -------------------------------------------------------------------------------- /config/dataset/generate_init_obs_libero.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/dataset/generate_init_obs_libero.yaml -------------------------------------------------------------------------------- /config/dataset/preprocess_calvin.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/dataset/preprocess_calvin.yaml -------------------------------------------------------------------------------- /config/dataset/preprocess_calvin_state.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/dataset/preprocess_calvin_state.yaml -------------------------------------------------------------------------------- /config/dataset/preprocess_libero.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/dataset/preprocess_libero.yaml -------------------------------------------------------------------------------- /config/dataset/regenerate_libero.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/dataset/regenerate_libero.yaml -------------------------------------------------------------------------------- /config/dataset/split_calvin.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/dataset/split_calvin.yaml -------------------------------------------------------------------------------- /config/env/calvin.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/env/calvin.yaml -------------------------------------------------------------------------------- /config/featurizer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/featurizer.yaml -------------------------------------------------------------------------------- /config/featurizer_libero.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/featurizer_libero.yaml -------------------------------------------------------------------------------- /config/libero/finetune/kitchen_scene1/ft_mb_ppo_diffusion_mlp_feat_vision.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/libero/finetune/kitchen_scene1/ft_mb_ppo_diffusion_mlp_feat_vision.yaml -------------------------------------------------------------------------------- /config/libero/pretrain/kitchen_scene1/pre_diffusion_mlp_feat_vision.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/libero/pretrain/kitchen_scene1/pre_diffusion_mlp_feat_vision.yaml -------------------------------------------------------------------------------- /config/libero/pretrain/kitchen_scene1/pre_diffusion_mlp_img.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/libero/pretrain/kitchen_scene1/pre_diffusion_mlp_img.yaml -------------------------------------------------------------------------------- /config/libero/pretrain/kitchen_scene1/pre_diffusion_mlp_state.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/libero/pretrain/kitchen_scene1/pre_diffusion_mlp_state.yaml -------------------------------------------------------------------------------- /config/logger/none.yaml: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /config/logger/wandb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/logger/wandb.yaml -------------------------------------------------------------------------------- /config/rewcls/balance_rewcls_dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/rewcls/balance_rewcls_dataset.yaml -------------------------------------------------------------------------------- /config/rewcls/balance_rewcls_dataset_libero.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/rewcls/balance_rewcls_dataset_libero.yaml -------------------------------------------------------------------------------- /config/rewcls/contrastive.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/rewcls/contrastive.yaml -------------------------------------------------------------------------------- /config/rewcls/contrastive_img.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/rewcls/contrastive_img.yaml -------------------------------------------------------------------------------- /config/rewcls/contrastive_libero.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/rewcls/contrastive_libero.yaml -------------------------------------------------------------------------------- /config/rewcls/generate_rewards.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/rewcls/generate_rewards.yaml -------------------------------------------------------------------------------- /config/rewcls/generate_rewards_libero.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/rewcls/generate_rewards_libero.yaml -------------------------------------------------------------------------------- /config/rewcls/rewcls_dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/rewcls/rewcls_dataset.yaml -------------------------------------------------------------------------------- /config/rewcls/rewcls_dataset_libero.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/rewcls/rewcls_dataset_libero.yaml -------------------------------------------------------------------------------- /config/rewcls/rewcls_img_dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/rewcls/rewcls_img_dataset.yaml -------------------------------------------------------------------------------- /config/setup.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/setup.yaml -------------------------------------------------------------------------------- /config/train_hybrid_wm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/train_hybrid_wm.yaml -------------------------------------------------------------------------------- /config/train_state_wm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/train_state_wm.yaml -------------------------------------------------------------------------------- /config/train_wm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/train_wm.yaml -------------------------------------------------------------------------------- /config/train_wm_libero.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/config/train_wm_libero.yaml -------------------------------------------------------------------------------- /dataset/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/dataset/README.md -------------------------------------------------------------------------------- /dataset/download_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/dataset/download_data.sh -------------------------------------------------------------------------------- /diwa/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/__init__.py -------------------------------------------------------------------------------- /diwa/agent/eval/eval_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/agent/eval/eval_agent.py -------------------------------------------------------------------------------- /diwa/agent/eval/eval_diffusion_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/agent/eval/eval_diffusion_agent.py -------------------------------------------------------------------------------- /diwa/agent/eval/eval_diffusion_img_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/agent/eval/eval_diffusion_img_agent.py -------------------------------------------------------------------------------- /diwa/agent/finetune/train_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/agent/finetune/train_agent.py -------------------------------------------------------------------------------- /diwa/agent/finetune/train_mb_ppo_diffusion_agent_visionwm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/agent/finetune/train_mb_ppo_diffusion_agent_visionwm.py -------------------------------------------------------------------------------- /diwa/agent/finetune/train_ppo_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/agent/finetune/train_ppo_agent.py -------------------------------------------------------------------------------- /diwa/agent/finetune/train_ppo_diffusion_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/agent/finetune/train_ppo_diffusion_agent.py -------------------------------------------------------------------------------- /diwa/agent/finetune/train_ppo_diffusion_img_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/agent/finetune/train_ppo_diffusion_img_agent.py -------------------------------------------------------------------------------- /diwa/agent/pretrain/train_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/agent/pretrain/train_agent.py -------------------------------------------------------------------------------- /diwa/agent/pretrain/train_diffusion_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/agent/pretrain/train_diffusion_agent.py -------------------------------------------------------------------------------- /diwa/agent/pretrain/train_gaussian_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/agent/pretrain/train_gaussian_agent.py -------------------------------------------------------------------------------- /diwa/dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /diwa/dataset/rewcls_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/dataset/rewcls_dataset.py -------------------------------------------------------------------------------- /diwa/dataset/sequence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/dataset/sequence.py -------------------------------------------------------------------------------- /diwa/env/runner/base_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/env/runner/base_runner.py -------------------------------------------------------------------------------- /diwa/env/runner/hybridwm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/env/runner/hybridwm.py -------------------------------------------------------------------------------- /diwa/env/runner/hybridwm_gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/env/runner/hybridwm_gaussian.py -------------------------------------------------------------------------------- /diwa/env/runner/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/env/runner/image.py -------------------------------------------------------------------------------- /diwa/env/runner/statewm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/env/runner/statewm.py -------------------------------------------------------------------------------- /diwa/env/runner/visionwm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/env/runner/visionwm.py -------------------------------------------------------------------------------- /diwa/env/runner/visionwm_gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/env/runner/visionwm_gaussian.py -------------------------------------------------------------------------------- /diwa/env/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/env/utils/__init__.py -------------------------------------------------------------------------------- /diwa/env/utils/calvin_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/env/utils/calvin_helpers.py -------------------------------------------------------------------------------- /diwa/env/utils/custom_subprocvec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/env/utils/custom_subprocvec.py -------------------------------------------------------------------------------- /diwa/env/utils/custom_subprocvec_libero.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/env/utils/custom_subprocvec_libero.py -------------------------------------------------------------------------------- /diwa/env/wrapper/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /diwa/env/wrapper/base_calvin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/env/wrapper/base_calvin.py -------------------------------------------------------------------------------- /diwa/env/wrapper/base_libero.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/env/wrapper/base_libero.py -------------------------------------------------------------------------------- /diwa/env/wrapper/calvin_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/env/wrapper/calvin_image.py -------------------------------------------------------------------------------- /diwa/env/wrapper/calvin_image_stacked.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/env/wrapper/calvin_image_stacked.py -------------------------------------------------------------------------------- /diwa/env/wrapper/calvin_image_stateful.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/env/wrapper/calvin_image_stateful.py -------------------------------------------------------------------------------- /diwa/env/wrapper/calvin_lowdim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/env/wrapper/calvin_lowdim.py -------------------------------------------------------------------------------- /diwa/env/wrapper/libero_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/env/wrapper/libero_image.py -------------------------------------------------------------------------------- /diwa/env/wrapper/libero_image_stacked.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/env/wrapper/libero_image_stacked.py -------------------------------------------------------------------------------- /diwa/env/wrapper/libero_image_stateful.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/env/wrapper/libero_image_stateful.py -------------------------------------------------------------------------------- /diwa/env/wrapper/libero_lowdim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/env/wrapper/libero_lowdim.py -------------------------------------------------------------------------------- /diwa/env/wrapper/multistep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/env/wrapper/multistep.py -------------------------------------------------------------------------------- /diwa/model/common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /diwa/model/common/critic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/model/common/critic.py -------------------------------------------------------------------------------- /diwa/model/common/gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/model/common/gaussian.py -------------------------------------------------------------------------------- /diwa/model/common/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/model/common/mlp.py -------------------------------------------------------------------------------- /diwa/model/common/mlp_gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/model/common/mlp_gaussian.py -------------------------------------------------------------------------------- /diwa/model/common/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/model/common/modules.py -------------------------------------------------------------------------------- /diwa/model/common/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/model/common/transformer.py -------------------------------------------------------------------------------- /diwa/model/common/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/model/common/vit.py -------------------------------------------------------------------------------- /diwa/model/diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /diwa/model/diffusion/diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/model/diffusion/diffusion.py -------------------------------------------------------------------------------- /diwa/model/diffusion/diffusion_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/model/diffusion/diffusion_eval.py -------------------------------------------------------------------------------- /diwa/model/diffusion/diffusion_ppo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/model/diffusion/diffusion_ppo.py -------------------------------------------------------------------------------- /diwa/model/diffusion/diffusion_vpg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/model/diffusion/diffusion_vpg.py -------------------------------------------------------------------------------- /diwa/model/diffusion/eta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/model/diffusion/eta.py -------------------------------------------------------------------------------- /diwa/model/diffusion/mlp_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/model/diffusion/mlp_diffusion.py -------------------------------------------------------------------------------- /diwa/model/diffusion/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/model/diffusion/modules.py -------------------------------------------------------------------------------- /diwa/model/diffusion/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/model/diffusion/sampling.py -------------------------------------------------------------------------------- /diwa/model/diffusion/sde_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/model/diffusion/sde_lib.py -------------------------------------------------------------------------------- /diwa/model/diffusion/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/model/diffusion/unet.py -------------------------------------------------------------------------------- /diwa/model/gaussian/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /diwa/model/gaussian/gaussian_ppo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/model/gaussian/gaussian_ppo.py -------------------------------------------------------------------------------- /diwa/model/gaussian/gaussian_vpg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/model/gaussian/gaussian_vpg.py -------------------------------------------------------------------------------- /diwa/model/rewcls/contrastive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/model/rewcls/contrastive.py -------------------------------------------------------------------------------- /diwa/model/rewcls/resnet_contrastive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/model/rewcls/resnet_contrastive.py -------------------------------------------------------------------------------- /diwa/utils/config_fix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/utils/config_fix.py -------------------------------------------------------------------------------- /diwa/utils/pytorch3d_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/utils/pytorch3d_transforms.py -------------------------------------------------------------------------------- /diwa/utils/reward_scaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/utils/reward_scaling.py -------------------------------------------------------------------------------- /diwa/utils/rotation_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/utils/rotation_transformer.py -------------------------------------------------------------------------------- /diwa/utils/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/utils/scheduler.py -------------------------------------------------------------------------------- /diwa/utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/utils/timer.py -------------------------------------------------------------------------------- /diwa/utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/utils/transforms.py -------------------------------------------------------------------------------- /diwa/wm/encoder/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/wm/encoder/base.py -------------------------------------------------------------------------------- /diwa/wm/encoder/hybridwm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/wm/encoder/hybridwm.py -------------------------------------------------------------------------------- /diwa/wm/encoder/statewm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/wm/encoder/statewm.py -------------------------------------------------------------------------------- /diwa/wm/encoder/visionwm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/wm/encoder/visionwm.py -------------------------------------------------------------------------------- /diwa/wm/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/wm/utils.py -------------------------------------------------------------------------------- /diwa/wm/wrapper/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/wm/wrapper/base.py -------------------------------------------------------------------------------- /diwa/wm/wrapper/hybridwm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/wm/wrapper/hybridwm.py -------------------------------------------------------------------------------- /diwa/wm/wrapper/hybridwm_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/wm/wrapper/hybridwm_cls.py -------------------------------------------------------------------------------- /diwa/wm/wrapper/statewm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/wm/wrapper/statewm.py -------------------------------------------------------------------------------- /diwa/wm/wrapper/visionwm_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/diwa/wm/wrapper/visionwm_cls.py -------------------------------------------------------------------------------- /docs/diwa_cover.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/docs/diwa_cover.png -------------------------------------------------------------------------------- /docs/diwa_overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/docs/diwa_overview.png -------------------------------------------------------------------------------- /install.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/install.sh -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements-dev.txt: -------------------------------------------------------------------------------- 1 | black 2 | flake8 3 | pre-commit -------------------------------------------------------------------------------- /requirements-libero.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/requirements-libero.txt -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/dataset/calculate_meanstd_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/scripts/dataset/calculate_meanstd_stats.py -------------------------------------------------------------------------------- /scripts/dataset/calculate_minmax_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/scripts/dataset/calculate_minmax_stats.py -------------------------------------------------------------------------------- /scripts/dataset/convert_libero_to_calvin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/scripts/dataset/convert_libero_to_calvin.py -------------------------------------------------------------------------------- /scripts/dataset/extract_expert_demos_feat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/scripts/dataset/extract_expert_demos_feat.py -------------------------------------------------------------------------------- /scripts/dataset/extract_expert_demos_img.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/scripts/dataset/extract_expert_demos_img.py -------------------------------------------------------------------------------- /scripts/dataset/extract_expert_demos_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/scripts/dataset/extract_expert_demos_state.py -------------------------------------------------------------------------------- /scripts/dataset/generate_init_obs_libero.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/scripts/dataset/generate_init_obs_libero.py -------------------------------------------------------------------------------- /scripts/dataset/preprocess_calvin_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/scripts/dataset/preprocess_calvin_dataset.py -------------------------------------------------------------------------------- /scripts/dataset/regenerate_libero_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/scripts/dataset/regenerate_libero_dataset.py -------------------------------------------------------------------------------- /scripts/dataset/split_calvin_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/scripts/dataset/split_calvin_dataset.py -------------------------------------------------------------------------------- /scripts/featurizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/scripts/featurizer.py -------------------------------------------------------------------------------- /scripts/rewcls/balance_rewcls_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/scripts/rewcls/balance_rewcls_dataset.py -------------------------------------------------------------------------------- /scripts/rewcls/balance_rewcls_img_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/scripts/rewcls/balance_rewcls_img_dataset.py -------------------------------------------------------------------------------- /scripts/rewcls/create_rewcls_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/scripts/rewcls/create_rewcls_dataset.py -------------------------------------------------------------------------------- /scripts/rewcls/create_rewcls_img_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/scripts/rewcls/create_rewcls_img_dataset.py -------------------------------------------------------------------------------- /scripts/rewcls/generate_rewards_calvin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/scripts/rewcls/generate_rewards_calvin.py -------------------------------------------------------------------------------- /scripts/rewcls/generate_rewards_libero.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/scripts/rewcls/generate_rewards_libero.py -------------------------------------------------------------------------------- /scripts/rewcls/train_contrastive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/scripts/rewcls/train_contrastive.py -------------------------------------------------------------------------------- /scripts/rewcls/train_img_contrastive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/scripts/rewcls/train_img_contrastive.py -------------------------------------------------------------------------------- /scripts/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/scripts/run.py -------------------------------------------------------------------------------- /scripts/set_path.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/scripts/set_path.sh -------------------------------------------------------------------------------- /scripts/tests/visionwm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/scripts/tests/visionwm.py -------------------------------------------------------------------------------- /scripts/train_wm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/scripts/train_wm.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/acl21/diwa/HEAD/setup.py --------------------------------------------------------------------------------