├── LICENSE ├── README.md ├── data └── data.txt ├── environment.yml ├── maniskill2_patches ├── peg_insertion_side.py ├── pick_cube.py ├── record.py └── turn_faucet.py ├── scripts ├── eval.sh ├── replay_turn_faucet_trajectories.sh └── train.sh └── src ├── concept_eval.py ├── concept_train.py ├── data.py ├── eval.py ├── infer_manisk_nond.py ├── model.py ├── path.py ├── perceiver ├── __init__.py ├── attention.py ├── attention_block.py ├── decoder.py ├── encoder.py ├── gauss_mild.py ├── perceiver.py ├── perceiver_lap.py ├── perceiver_sig.py ├── query.py └── query_new.py ├── train.py ├── train_utils.py ├── util ├── epoch_timer.py └── look_table.py └── vec_env.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/README.md -------------------------------------------------------------------------------- /data/data.txt: -------------------------------------------------------------------------------- 1 | Download the dataset from Google Drive and put it here. -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/environment.yml -------------------------------------------------------------------------------- /maniskill2_patches/peg_insertion_side.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/maniskill2_patches/peg_insertion_side.py -------------------------------------------------------------------------------- /maniskill2_patches/pick_cube.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/maniskill2_patches/pick_cube.py -------------------------------------------------------------------------------- /maniskill2_patches/record.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/maniskill2_patches/record.py -------------------------------------------------------------------------------- /maniskill2_patches/turn_faucet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/maniskill2_patches/turn_faucet.py -------------------------------------------------------------------------------- /scripts/eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/scripts/eval.sh -------------------------------------------------------------------------------- /scripts/replay_turn_faucet_trajectories.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/scripts/replay_turn_faucet_trajectories.sh -------------------------------------------------------------------------------- /scripts/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/scripts/train.sh -------------------------------------------------------------------------------- /src/concept_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/src/concept_eval.py -------------------------------------------------------------------------------- /src/concept_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/src/concept_train.py -------------------------------------------------------------------------------- /src/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/src/data.py -------------------------------------------------------------------------------- /src/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/src/eval.py -------------------------------------------------------------------------------- /src/infer_manisk_nond.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/src/infer_manisk_nond.py -------------------------------------------------------------------------------- /src/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/src/model.py -------------------------------------------------------------------------------- /src/path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/src/path.py -------------------------------------------------------------------------------- /src/perceiver/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/perceiver/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/src/perceiver/attention.py -------------------------------------------------------------------------------- /src/perceiver/attention_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/src/perceiver/attention_block.py -------------------------------------------------------------------------------- /src/perceiver/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/src/perceiver/decoder.py -------------------------------------------------------------------------------- /src/perceiver/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/src/perceiver/encoder.py -------------------------------------------------------------------------------- /src/perceiver/gauss_mild.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/src/perceiver/gauss_mild.py -------------------------------------------------------------------------------- /src/perceiver/perceiver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/src/perceiver/perceiver.py -------------------------------------------------------------------------------- /src/perceiver/perceiver_lap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/src/perceiver/perceiver_lap.py -------------------------------------------------------------------------------- /src/perceiver/perceiver_sig.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/src/perceiver/perceiver_sig.py -------------------------------------------------------------------------------- /src/perceiver/query.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/src/perceiver/query.py -------------------------------------------------------------------------------- /src/perceiver/query_new.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/src/perceiver/query_new.py -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/src/train.py -------------------------------------------------------------------------------- /src/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/src/train_utils.py -------------------------------------------------------------------------------- /src/util/epoch_timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/src/util/epoch_timer.py -------------------------------------------------------------------------------- /src/util/look_table.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/src/util/look_table.py -------------------------------------------------------------------------------- /src/vec_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeiZhou26/MaxMI/HEAD/src/vec_env.py --------------------------------------------------------------------------------