Code for Environment Dynamics Decomposition (ED2).
Follow the installation in MBPO and Dreamer.
First follow the SD2 method for sub-dynamics discovery (we introduced it in SD2/README.md). Then follow the SD2 result, run the experiment in ED2-Dreamer or ED2-MBPO (following the detail operation in ED2-Dreamer/README.md or ED2-MBPO/README.md).