pravsels's picture
Add model card with W&B link
44d06fb verified
|
Raw
History Blame Contribute Delete
1.14 kB
metadata
library_name: gr00t
tags:
  - gr00t
  - gr00t-n1.7
  - robotics
  - vla
base_model: nvidia/GR00T-N1.7-3B

groot1.7_cable_unclip

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

Base model nvidia/GR00T-N1.7-3B
Dataset pravsels/cable_unclip_remote
Task cable_unclip
Training Isambard-AI GH200, batch 64, action horizon 16
W&B project groot1.7_cable_unclip
W&B run oxz2z06w

Checkpoints

Step Loss Path
8,000 0.0233 checkpoints/8000/pretrained_model/
6,000 0.0282 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_cable_unclip/checkpoints/8000/pretrained_model"