This repository contains a TensorFlow implementation of [CDA-MBPO]
- Install MuJoCo 1.50 at
~/.mujoco/mjpro150
and copy your license key to~/.mujoco/mjkey.txt
pip install -r requirements.txt
Configuration files can be found in config/
.
Run
python main.py --config=config.ant
The code implementation is mainly modified based on the MBPO and AMPO codebase.