This is the official repo of "Causal Dynamics Learning for Task-Independent State Abstraction" at ICML 2022.
To run the code, modify parameters in cdl/policy_params.json
and then run main_policy.py
in cdl
directory. Example parameter files are in example_params
.
The customized robosuite environment can be found in: https://github.com/wangzizhao/robosuite/tree/cdl.