| 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](https://huggingface.co/nvidia/GR00T-N1.7-3B) on SO101 data (`cable_unclip`). | |
| | | | | |
| |---|---| | |
| | **Base model** | [nvidia/GR00T-N1.7-3B](https://huggingface.co/nvidia/GR00T-N1.7-3B) | | |
| | **Dataset** | [pravsels/cable_unclip_remote](https://huggingface.co/datasets/pravsels/cable_unclip_remote) | | |
| | **Task** | `cable_unclip` | | |
| | **Training** | Isambard-AI GH200, batch 64, action horizon 16 | | |
| | **W&B project** | [groot1.7_cable_unclip](https://wandb.ai/pravsels/groot1.7_cable_unclip) | | |
| | **W&B run** | [oxz2z06w](https://wandb.ai/pravsels/groot1.7_cable_unclip/runs/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: | |
| ```python | |
| from pathlib import Path | |
| # Use checkpoints/<step>/pretrained_model/ with the GR00T inference stack | |
| ckpt = "pravsels/groot1.7_cable_unclip/checkpoints/8000/pretrained_model" | |
| ``` | |