├── .gitignore ├── LICENSE ├── README.md ├── cfgs ├── __init__.py └── prise_config.yaml ├── convert_data.py ├── eval_libero90.py ├── images └── prise_demo.mp4 ├── prise_agent.py ├── replay_buffer.py ├── requirements.txt ├── setup.py ├── tokenizer_api.py ├── train_prise.py └── utils ├── data_augmentation.py ├── libero_wrapper.py ├── logger.py ├── misc.py ├── policy_head.py ├── quantizer.py └── resnet18_encoder.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FrankZheng2022/PRISE/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FrankZheng2022/PRISE/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FrankZheng2022/PRISE/HEAD/README.md -------------------------------------------------------------------------------- /cfgs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cfgs/prise_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FrankZheng2022/PRISE/HEAD/cfgs/prise_config.yaml -------------------------------------------------------------------------------- /convert_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FrankZheng2022/PRISE/HEAD/convert_data.py -------------------------------------------------------------------------------- /eval_libero90.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FrankZheng2022/PRISE/HEAD/eval_libero90.py -------------------------------------------------------------------------------- /images/prise_demo.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FrankZheng2022/PRISE/HEAD/images/prise_demo.mp4 -------------------------------------------------------------------------------- /prise_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FrankZheng2022/PRISE/HEAD/prise_agent.py -------------------------------------------------------------------------------- /replay_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FrankZheng2022/PRISE/HEAD/replay_buffer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FrankZheng2022/PRISE/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FrankZheng2022/PRISE/HEAD/setup.py -------------------------------------------------------------------------------- /tokenizer_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FrankZheng2022/PRISE/HEAD/tokenizer_api.py -------------------------------------------------------------------------------- /train_prise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FrankZheng2022/PRISE/HEAD/train_prise.py -------------------------------------------------------------------------------- /utils/data_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FrankZheng2022/PRISE/HEAD/utils/data_augmentation.py -------------------------------------------------------------------------------- /utils/libero_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FrankZheng2022/PRISE/HEAD/utils/libero_wrapper.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FrankZheng2022/PRISE/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FrankZheng2022/PRISE/HEAD/utils/misc.py -------------------------------------------------------------------------------- /utils/policy_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FrankZheng2022/PRISE/HEAD/utils/policy_head.py -------------------------------------------------------------------------------- /utils/quantizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FrankZheng2022/PRISE/HEAD/utils/quantizer.py -------------------------------------------------------------------------------- /utils/resnet18_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FrankZheng2022/PRISE/HEAD/utils/resnet18_encoder.py --------------------------------------------------------------------------------