groot1.7_block_stack_20k

Fine-tuned GR00T N1.7-3B for block_stack on SO101 data.

Base model nvidia/GR00T-N1.7-3B
Dataset villekuosmanen/armnetbench_block_stack
Task block_stack
Embodiment single-arm SO-101
Training 20k steps, batch 32 (8/GPU × 4 DDP), 4× RTX 5090 on Vast.ai, load_bf16=True
Prior HF repo pravsels/groot1.7_block_stack
W&B project groot1.7_block_stack
W&B run pwazfb2z

Checkpoints

Final checkpoint (step 20000) lives at the repository root for direct loading.

Usage

Load with the GR00T inference stack from this repo root (weights + processor configs).

Downloads last month
2
Safetensors
Model size
3B params
Tensor type
BF16
·
Video Preview
loading

Model tree for pravsels/groot1.7_block_stack_20k

Finetuned
(61)
this model