This repository implements the Bigger, Better, Faster (BBF) agent in JAX, building on Dopamine. SPR (Schwarzer et al, 2021) and SR-SPR (D'Oro et al, 2023) may also be run as hyperparameter configurations.
To install the repository, simply run pip install -r requirements.txt
.
Note that depending on your operating system and cuda version extra steps may be necessary to
successfully install JAX: please see the JAX install instructions for guidance.
To run a BBF agent locally, run
python -m bbf.train \
--agent=BBF \
--gin_files=bbf/configs/BBF.gin \
--base_dir=/tmp/online_rl/bbf \
--run_number=1