Upload policy weights, train config and readme
Browse files- README.md +1 -1
- config.json +20 -2
- model.safetensors +1 -1
- train_config.json +24 -6
README.md
CHANGED
|
@@ -7,8 +7,8 @@ model_name: smolvla
|
|
| 7 |
pipeline_tag: robotics
|
| 8 |
tags:
|
| 9 |
- lerobot
|
| 10 |
-
- smolvla
|
| 11 |
- robotics
|
|
|
|
| 12 |
---
|
| 13 |
|
| 14 |
# Model Card for smolvla
|
|
|
|
| 7 |
pipeline_tag: robotics
|
| 8 |
tags:
|
| 9 |
- lerobot
|
|
|
|
| 10 |
- robotics
|
| 11 |
+
- smolvla
|
| 12 |
---
|
| 13 |
|
| 14 |
# Model Card for smolvla
|
config.json
CHANGED
|
@@ -2,7 +2,7 @@
|
|
| 2 |
"type": "smolvla",
|
| 3 |
"n_obs_steps": 1,
|
| 4 |
"input_features": {
|
| 5 |
-
"observation.images.
|
| 6 |
"type": "VISUAL",
|
| 7 |
"shape": [
|
| 8 |
256,
|
|
@@ -10,7 +10,7 @@
|
|
| 10 |
3
|
| 11 |
]
|
| 12 |
},
|
| 13 |
-
"observation.images.
|
| 14 |
"type": "VISUAL",
|
| 15 |
"shape": [
|
| 16 |
256,
|
|
@@ -23,6 +23,24 @@
|
|
| 23 |
"shape": [
|
| 24 |
8
|
| 25 |
]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
}
|
| 27 |
},
|
| 28 |
"output_features": {
|
|
|
|
| 2 |
"type": "smolvla",
|
| 3 |
"n_obs_steps": 1,
|
| 4 |
"input_features": {
|
| 5 |
+
"observation.images.image": {
|
| 6 |
"type": "VISUAL",
|
| 7 |
"shape": [
|
| 8 |
256,
|
|
|
|
| 10 |
3
|
| 11 |
]
|
| 12 |
},
|
| 13 |
+
"observation.images.wrist_image": {
|
| 14 |
"type": "VISUAL",
|
| 15 |
"shape": [
|
| 16 |
256,
|
|
|
|
| 23 |
"shape": [
|
| 24 |
8
|
| 25 |
]
|
| 26 |
+
},
|
| 27 |
+
"observation.states.ee_state": {
|
| 28 |
+
"type": "STATE",
|
| 29 |
+
"shape": [
|
| 30 |
+
6
|
| 31 |
+
]
|
| 32 |
+
},
|
| 33 |
+
"observation.states.joint_state": {
|
| 34 |
+
"type": "STATE",
|
| 35 |
+
"shape": [
|
| 36 |
+
7
|
| 37 |
+
]
|
| 38 |
+
},
|
| 39 |
+
"observation.states.gripper_state": {
|
| 40 |
+
"type": "STATE",
|
| 41 |
+
"shape": [
|
| 42 |
+
2
|
| 43 |
+
]
|
| 44 |
}
|
| 45 |
},
|
| 46 |
"output_features": {
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 1197789224
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:84d71cc4cff56f9169a1a1553d43d59885e59b4a3d862b6433db1a3d83ea7abb
|
| 3 |
size 1197789224
|
train_config.json
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
{
|
| 2 |
"dataset": {
|
| 3 |
"repo_id": "none",
|
| 4 |
-
"root": "/mnt/data/sftp/data/tannq3/data/IPEC-COMMUNITY/
|
| 5 |
"episodes": null,
|
| 6 |
"image_transforms": {
|
| 7 |
"enable": false,
|
|
@@ -84,7 +84,7 @@
|
|
| 84 |
"type": "smolvla",
|
| 85 |
"n_obs_steps": 1,
|
| 86 |
"input_features": {
|
| 87 |
-
"observation.images.
|
| 88 |
"type": "VISUAL",
|
| 89 |
"shape": [
|
| 90 |
256,
|
|
@@ -92,7 +92,7 @@
|
|
| 92 |
3
|
| 93 |
]
|
| 94 |
},
|
| 95 |
-
"observation.images.
|
| 96 |
"type": "VISUAL",
|
| 97 |
"shape": [
|
| 98 |
256,
|
|
@@ -105,6 +105,24 @@
|
|
| 105 |
"shape": [
|
| 106 |
8
|
| 107 |
]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 108 |
}
|
| 109 |
},
|
| 110 |
"output_features": {
|
|
@@ -171,8 +189,8 @@
|
|
| 171 |
"max_period": 4.0,
|
| 172 |
"rtc_config": null
|
| 173 |
},
|
| 174 |
-
"output_dir": "/mnt/data/sftp/data/tannq3/workspace/outputs/
|
| 175 |
-
"job_name": "
|
| 176 |
"resume": false,
|
| 177 |
"seed": 1000,
|
| 178 |
"num_workers": 4,
|
|
@@ -213,7 +231,7 @@
|
|
| 213 |
"project": "CoRL26_Equi",
|
| 214 |
"entity": null,
|
| 215 |
"notes": null,
|
| 216 |
-
"run_id": "
|
| 217 |
"mode": null
|
| 218 |
},
|
| 219 |
"peft": null,
|
|
|
|
| 1 |
{
|
| 2 |
"dataset": {
|
| 3 |
"repo_id": "none",
|
| 4 |
+
"root": "/mnt/data/sftp/data/tannq3/data/IPEC-COMMUNITY/libero_90_no_noops_lerobot",
|
| 5 |
"episodes": null,
|
| 6 |
"image_transforms": {
|
| 7 |
"enable": false,
|
|
|
|
| 84 |
"type": "smolvla",
|
| 85 |
"n_obs_steps": 1,
|
| 86 |
"input_features": {
|
| 87 |
+
"observation.images.image": {
|
| 88 |
"type": "VISUAL",
|
| 89 |
"shape": [
|
| 90 |
256,
|
|
|
|
| 92 |
3
|
| 93 |
]
|
| 94 |
},
|
| 95 |
+
"observation.images.wrist_image": {
|
| 96 |
"type": "VISUAL",
|
| 97 |
"shape": [
|
| 98 |
256,
|
|
|
|
| 105 |
"shape": [
|
| 106 |
8
|
| 107 |
]
|
| 108 |
+
},
|
| 109 |
+
"observation.states.ee_state": {
|
| 110 |
+
"type": "STATE",
|
| 111 |
+
"shape": [
|
| 112 |
+
6
|
| 113 |
+
]
|
| 114 |
+
},
|
| 115 |
+
"observation.states.joint_state": {
|
| 116 |
+
"type": "STATE",
|
| 117 |
+
"shape": [
|
| 118 |
+
7
|
| 119 |
+
]
|
| 120 |
+
},
|
| 121 |
+
"observation.states.gripper_state": {
|
| 122 |
+
"type": "STATE",
|
| 123 |
+
"shape": [
|
| 124 |
+
2
|
| 125 |
+
]
|
| 126 |
}
|
| 127 |
},
|
| 128 |
"output_features": {
|
|
|
|
| 189 |
"max_period": 4.0,
|
| 190 |
"rtc_config": null
|
| 191 |
},
|
| 192 |
+
"output_dir": "/mnt/data/sftp/data/tannq3/workspace/outputs/smolvla_libero_90_steps30k_bsz64_2026-04-23_22-25-51",
|
| 193 |
+
"job_name": "smolvla_libero_90_steps30k_bsz64_2026-04-23_22-25-51",
|
| 194 |
"resume": false,
|
| 195 |
"seed": 1000,
|
| 196 |
"num_workers": 4,
|
|
|
|
| 231 |
"project": "CoRL26_Equi",
|
| 232 |
"entity": null,
|
| 233 |
"notes": null,
|
| 234 |
+
"run_id": "w6mxzkb4",
|
| 235 |
"mode": null
|
| 236 |
},
|
| 237 |
"peft": null,
|