Skip to content

Latest commit

 

History

History
 
 

bigger_better_faster

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 

Bigger, Better, Faster: Human Level Atari with Human Level Efficiency

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.

Setup

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.

Training

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

References