Tuning Legged Locomotion Controllers via Safe Bayesian Optimization

This repository contains the code for our paper Tuning Legged Locomotion Controllers via Safe Bayesian Optimization. A demo video is available at https://www.youtube.com/watch?v=pceHWpFr3ng.

All relevant code for the paper is in gosafeopt/aquisitions/*, gosafeopt/models/__init__.py, gosafeopt/optim/swarm_opt.py and gosafeopt/optim/base_optimizer.py.

Setup

#With poetry
poetry install

#With pip/venv
python -m venv .venv
. .venv/bin/activate
pip install -e .

It might be necessary to create a wandb account at wandb.ai if not already existing.

Examples

To train the pendulum toy problem model with a specific aquisition function and show the result plot run

poetry install --with examples
python examples/pendulum.py