visually compare robot reward functions
24
stars
14
commits
Python
primary language
Jul 13, 2026
updated
Reward functions you can run on your videos:
...and easy to add more!
Create an MP4 video of robot manipulation (example). For efficiency, please downsize the file to 480p, as image pixels are passed as tokens.
Install prerequisites
virtualenv venv
. venv/bin/activate
pip install -r requirements.txt
pip install torch torchvision transformers accelerate qwen-vl-utils
Run the reward algorithms on your video:
A. Run bruteforce_vlm, gvl, robodopamine, roboreward, and/or topreward
Run the script to calculate reward functions on your video:
python run_rewards.py --method bruteforce_vlm,gvl,robodopamine,roboreward,topreward --video <myvideo.mp4> --instruction <instructions e.g. `create a tower of 5 cubes`>
Notes:
gvl or bruteforce_vlm, you must include an OpenAI API key:--openai-api-key <your key>
topreward, roboreward, and/or robodopamine, you'll need at least 16GB unified/GPU memory.B. Run robometer
Compute the reward using the custom script here.
View the results in your browser:
./run_viewer.sh
14 commits
Python
73.4%
HTML
26.5%
visually compare robot reward functions
24
stars
14
commits
Python
primary language
Jul 13, 2026
updated
Reward functions you can run on your videos:
...and easy to add more!
Create an MP4 video of robot manipulation (example). For efficiency, please downsize the file to 480p, as image pixels are passed as tokens.
Install prerequisites
virtualenv venv
. venv/bin/activate
pip install -r requirements.txt
pip install torch torchvision transformers accelerate qwen-vl-utils
Run the reward algorithms on your video:
A. Run bruteforce_vlm, gvl, robodopamine, roboreward, and/or topreward
Run the script to calculate reward functions on your video:
python run_rewards.py --method bruteforce_vlm,gvl,robodopamine,roboreward,topreward --video <myvideo.mp4> --instruction <instructions e.g. `create a tower of 5 cubes`>
Notes:
gvl or bruteforce_vlm, you must include an OpenAI API key:--openai-api-key <your key>
topreward, roboreward, and/or robodopamine, you'll need at least 16GB unified/GPU memory.B. Run robometer
Compute the reward using the custom script here.
View the results in your browser:
./run_viewer.sh
14 commits
Python
73.4%
HTML
26.5%