Benchmark results
Bridge dataset (WidowX robot)
Checkpoint: nvidia/GR00T-N1.6-bridge| Task | Success rate |
|---|---|
| widowx_spoon_on_towel | 129/200 (64.5%) |
| widowx_carrot_on_plate | 131/200 (65.5%) |
| widowx_put_eggplant_in_basket | 186/200 (93%) |
| widowx_stack_cube | 11/200 (5.5%) |
| widowx_put_eggplant_in_sink | 80/200 (40%) |
| widowx_close_drawer | 141/200 (70.5%) |
| widowx_open_drawer | 191/200 (95.5%) |
| Average | 62.07% |
Fractal dataset (Google robot)
Checkpoint: nvidia/GR00T-N1.6-fractal| Task | Success rate |
|---|---|
| google_robot_pick_coke_can | 195/200 (97.5%) |
| google_robot_pick_object | 174/200 (87%) |
| google_robot_move_near | 151/200 (75.5%) |
| google_robot_open_drawer | 88/200 (44%) |
| google_robot_close_drawer | 175/200 (87.5%) |
| google_robot_place_in_closed_drawer | 29/200 (14.5%) |
| Average | 67.66% |
Fine-tuning
Bridge dataset (WidowX robot)
Remember to set
WANDB_API_KEY if using Weights & Biases tracking, or remove the --use-wandb flag from the training script.Fractal dataset (Google robot)
Evaluation
Setup environment
Install the required dependencies (only needs to be done once):Run evaluation
Start policy server
In Terminal 1, choose one of the following options:Option 1: Local fine-tuned checkpointOption 2: Remote fine-tuned checkpoint
Available tasks
Google robot tasks
simpler_env_google/google_robot_pick_coke_cansimpler_env_google/google_robot_pick_objectsimpler_env_google/google_robot_move_nearsimpler_env_google/google_robot_open_drawersimpler_env_google/google_robot_close_drawersimpler_env_google/google_robot_place_in_closed_drawer
WidowX robot tasks
simpler_env_widowx/widowx_spoon_on_towelsimpler_env_widowx/widowx_carrot_on_platesimpler_env_widowx/widowx_put_eggplant_in_basketsimpler_env_widowx/widowx_stack_cubesimpler_env_widowx/widowx_put_eggplant_in_sinksimpler_env_widowx/widowx_close_drawersimpler_env_widowx/widowx_open_drawer