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"