groot1.7_cable_clip / README.md
pravsels's picture
Add model card with W&B link
117feda verified
|
Raw
History Blame Contribute Delete
1.13 kB
metadata
library_name: gr00t
tags:
  - gr00t
  - gr00t-n1.7
  - robotics
  - vla
base_model: nvidia/GR00T-N1.7-3B

groot1.7_cable_clip

Fine-tuned GR00T N1.7-3B on SO101 data (cable_clip).

Base model nvidia/GR00T-N1.7-3B
Dataset pravsels/cable_clip_remote_v2
Task cable_clip
Training Isambard-AI GH200, batch 64, action horizon 16
W&B project groot1.7_cable_clip
W&B run 2i1zpig0

Checkpoints

Step Loss Path
10,000 0.0264 checkpoints/10000/pretrained_model/
8,000 0.0279 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_cable_clip/checkpoints/10000/pretrained_model"