File size: 1,131 Bytes
117feda | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 | ---
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](https://huggingface.co/nvidia/GR00T-N1.7-3B) on SO101 data (`cable_clip`).
| | |
|---|---|
| **Base model** | [nvidia/GR00T-N1.7-3B](https://huggingface.co/nvidia/GR00T-N1.7-3B) |
| **Dataset** | [pravsels/cable_clip_remote_v2](https://huggingface.co/datasets/pravsels/cable_clip_remote_v2) |
| **Task** | `cable_clip` |
| **Training** | Isambard-AI GH200, batch 64, action horizon 16 |
| **W&B project** | [groot1.7_cable_clip](https://wandb.ai/pravsels/groot1.7_cable_clip) |
| **W&B run** | [2i1zpig0](https://wandb.ai/pravsels/groot1.7_cable_clip/runs/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:
```python
from pathlib import Path
# Use checkpoints/<step>/pretrained_model/ with the GR00T inference stack
ckpt = "pravsels/groot1.7_cable_clip/checkpoints/10000/pretrained_model"
```
|