├── LICENSE ├── README.md ├── environment.yml ├── plots ├── saqcql_kitchen.png └── saqiql_adroit.png ├── project_setup.bash ├── scripts ├── vqbc_gym.sh ├── vqcql_kitchen.sh └── vqiql_adroit.sh ├── unit1 └── 01_introduction_to_diffusers.ipynb └── vqn ├── __init__.py ├── brac.py ├── brac_main.py ├── conservative_sac.py ├── conservative_sac_main.py ├── jax_utils.py ├── model.py ├── replay_buffer.py ├── robomimic_utils.py ├── sac.py ├── sac_main.py ├── sampler.py ├── test_vqn.py ├── test_vqn_main.py ├── utils.py ├── vqiql.py ├── vqiql_main.py ├── vqn.py └── vqn_main.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/README.md -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/environment.yml -------------------------------------------------------------------------------- /plots/saqcql_kitchen.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/plots/saqcql_kitchen.png -------------------------------------------------------------------------------- /plots/saqiql_adroit.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/plots/saqiql_adroit.png -------------------------------------------------------------------------------- /project_setup.bash: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/project_setup.bash -------------------------------------------------------------------------------- /scripts/vqbc_gym.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/scripts/vqbc_gym.sh -------------------------------------------------------------------------------- /scripts/vqcql_kitchen.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/scripts/vqcql_kitchen.sh -------------------------------------------------------------------------------- /scripts/vqiql_adroit.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/scripts/vqiql_adroit.sh -------------------------------------------------------------------------------- /unit1/01_introduction_to_diffusers.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/unit1/01_introduction_to_diffusers.ipynb -------------------------------------------------------------------------------- /vqn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /vqn/brac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/vqn/brac.py -------------------------------------------------------------------------------- /vqn/brac_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/vqn/brac_main.py -------------------------------------------------------------------------------- /vqn/conservative_sac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/vqn/conservative_sac.py -------------------------------------------------------------------------------- /vqn/conservative_sac_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/vqn/conservative_sac_main.py -------------------------------------------------------------------------------- /vqn/jax_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/vqn/jax_utils.py -------------------------------------------------------------------------------- /vqn/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/vqn/model.py -------------------------------------------------------------------------------- /vqn/replay_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/vqn/replay_buffer.py -------------------------------------------------------------------------------- /vqn/robomimic_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/vqn/robomimic_utils.py -------------------------------------------------------------------------------- /vqn/sac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/vqn/sac.py -------------------------------------------------------------------------------- /vqn/sac_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/vqn/sac_main.py -------------------------------------------------------------------------------- /vqn/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/vqn/sampler.py -------------------------------------------------------------------------------- /vqn/test_vqn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/vqn/test_vqn.py -------------------------------------------------------------------------------- /vqn/test_vqn_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/vqn/test_vqn_main.py -------------------------------------------------------------------------------- /vqn/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/vqn/utils.py -------------------------------------------------------------------------------- /vqn/vqiql.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/vqn/vqiql.py -------------------------------------------------------------------------------- /vqn/vqiql_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/vqn/vqiql_main.py -------------------------------------------------------------------------------- /vqn/vqn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/vqn/vqn.py -------------------------------------------------------------------------------- /vqn/vqn_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jianlanluo/SAQ/HEAD/vqn/vqn_main.py --------------------------------------------------------------------------------