sufodsia/mbpo_pytorch

Python

MBPO PyTorch

A PyTorch reimplementation of MBPO (When to trust your model: model-based policy optimization)

Dependency

Please refer to ./requirements.txt.

Usage

pip install -e .

# default hyperparams in ./configs/mbpo.yaml
# remember to CHANGE proj_dir to your actual directory 
python ./mbpo_pytorch/scripts/run_mbpo.py

# you can also overwrite hyperparams by passing args, e.g.
python ./mbpo_pytorch/scripts/run_mbpo.py --set seed=0 verbose=1 device="'cuda:0'" env.env_name='FixedHopper'

Credits