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"
Downloads last month
22
Safetensors
Model size
3B params
Tensor type
F32
·
Video Preview
loading

Model tree for pravsels/groot1.7_tool_removal

Finetuned
(67)
this model