metadata
library_name: gr00t
tags:
- gr00t
- gr00t-n1.7
- robotics
- vla
base_model: nvidia/GR00T-N1.7-3B
groot1.7_tool_removal
Fine-tuned GR00T N1.7-3B on SO101 data (tool_removal).
| Base model | nvidia/GR00T-N1.7-3B |
| Dataset | villekuosmanen/armnetbench_tool_removal |
| Task | tool_removal |
| Training | Isambard-AI GH200, batch 64, action horizon 16 |
| W&B project | groot1.7_tool_removal |
| W&B run | v1mhfrxq |
Checkpoints
| Step | Loss | Path |
|---|---|---|
| 8,000 | 0.0291 | checkpoints/8000/pretrained_model/ |
| 6,000 | 0.0358 | checkpoints/6000/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_tool_removal/checkpoints/8000/pretrained_model"