metadata
library_name: gr00t
tags:
- gr00t
- gr00t-n1.7
- robotics
- vla
base_model: nvidia/GR00T-N1.7-3B
groot1.7_block_stack
Fine-tuned GR00T N1.7-3B on SO101 data (block_stack).
| Base model | nvidia/GR00T-N1.7-3B |
| Dataset | villekuosmanen/armnetbench_block_stack |
| Task | block_stack |
| Training | Isambard-AI GH200, batch 64, action horizon 16 |
| W&B project | groot1.7_block_stack |
| W&B run | ypeedac1 |
Checkpoints
| Step | Loss | Path |
|---|---|---|
| 10,000 | 0.0194 | checkpoints/10000/pretrained_model/ |
| 8,000 | 0.0262 | checkpoints/8000/pretrained_model/ |
Usage
Load from a specific step:
from pathlib import Path
# Use checkpoints/<step>/pretrained_model/ with the GR00T inference stack
ckpt = "pravsels/groot1.7_block_stack/checkpoints/10000/pretrained_model"