Downloads · 30 days
14
39% of all-time downloads
seungkukim/gr00t_n1_7_openarm_wam_v1_50k
gr00t_n1_7_openarm_wam_v1_50k is a robotics model from seungkukim. Use it for the robotics task on the model card, and read the license before you ship it in a product. It is set up for gr00t. The card lists the license as apache-2.0.
Fine-tune of nvidia/GR00T-N1.7-3B on the six robot subsets of openarmwamv1, a 28-dim bimanual OpenArm embodiment with two ego cameras.
Downloads · 30 days
14
39% of all-time downloads
All-time downloads
36
Public
Parameters
3.1B
6.9 GB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors6.9 GB · 100%
From the Hugging Face model README
Fine-tune of nvidia/GR00T-N1.7-3B
on the six robot subsets of openarm_wam_v1, a 28-dim bimanual OpenArm
embodiment with two ego cameras.
| Subset | Episodes | Frames | fps |
|---|---|---|---|
openarm_ego_jungwook | 151 | 58,023 | 30 |
openarm_teleop_v3/bottle | 96 | 35,506 | 20 |
openarm_teleop_v3/cup | 108 | 28,757 | 20 |
openarm_teleop_v3/doll | 48 | 13,414 | 20 |
openarm_teleop_v3/snack | 96 | 40,184 | 20 |
banana_v21_openarm28 | 54 | 16,202 | 30 |
| total | 553 | 192,086 |
Mixture weights are frame-proportional (len_i / sum(len)).
NEW_EMBODIMENT, 28-dim state/action--use-percentiles)--shortest-image-edge 192 --crop-fraction 0.95)Trained on 2×H100 80GB in 3 h 41 m. Final training loss 0.0195.
from gr00t.policy.gr00t_policy import Gr00tPolicy
policy = Gr00tPolicy(
model_path="seungkukim/gr00t_n1_7_openarm_wam_v1_50k",
embodiment_tag="new_embodiment",
)
action = policy.get_action(observation)
Or serve it:
python gr00t/eval/run_gr00t_server.py \
--model-path seungkukim/gr00t_n1_7_openarm_wam_v1_50k \
--embodiment-tag new_embodiment
Inference-ready weights, processor and configs. The DeepSpeed optimizer state is not included, so this repo cannot resume training.