Upload policy weights, train config and readme
Browse files- README.md +2 -2
- config.json +1 -1
- model.safetensors +1 -1
- train_config.json +3 -3
README.md
CHANGED
|
@@ -7,8 +7,8 @@ model_name: smolvla
|
|
| 7 |
pipeline_tag: robotics
|
| 8 |
tags:
|
| 9 |
- lerobot
|
| 10 |
-
- robotics
|
| 11 |
- smolvla
|
|
|
|
| 12 |
---
|
| 13 |
|
| 14 |
# Model Card for smolvla
|
|
@@ -86,7 +86,7 @@ The policy consumes these observation features and produces these action feature
|
|
| 86 |
|
| 87 |
| Setting | Value |
|
| 88 |
| --- | --- |
|
| 89 |
-
| Training steps |
|
| 90 |
| Batch size | 224 |
|
| 91 |
| Optimizer | adamw |
|
| 92 |
| Learning rate | 0.0001 |
|
|
|
|
| 7 |
pipeline_tag: robotics
|
| 8 |
tags:
|
| 9 |
- lerobot
|
|
|
|
| 10 |
- smolvla
|
| 11 |
+
- robotics
|
| 12 |
---
|
| 13 |
|
| 14 |
# Model Card for smolvla
|
|
|
|
| 86 |
|
| 87 |
| Setting | Value |
|
| 88 |
| --- | --- |
|
| 89 |
+
| Training steps | 2000 |
|
| 90 |
| Batch size | 224 |
|
| 91 |
| Optimizer | adamw |
|
| 92 |
| Learning rate | 0.0001 |
|
config.json
CHANGED
|
@@ -105,6 +105,6 @@
|
|
| 105 |
"min_period": 0.004,
|
| 106 |
"max_period": 4.0,
|
| 107 |
"rtc_config": null,
|
| 108 |
-
"compile_model":
|
| 109 |
"compile_mode": "max-autotune"
|
| 110 |
}
|
|
|
|
| 105 |
"min_period": 0.004,
|
| 106 |
"max_period": 4.0,
|
| 107 |
"rtc_config": null,
|
| 108 |
+
"compile_model": true,
|
| 109 |
"compile_mode": "max-autotune"
|
| 110 |
}
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 906712520
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:34a32220eb66320e357aac989553a7b979baecefb5565f34fd700b68a419208e
|
| 3 |
size 906712520
|
train_config.json
CHANGED
|
@@ -191,7 +191,7 @@
|
|
| 191 |
"min_period": 0.004,
|
| 192 |
"max_period": 4.0,
|
| 193 |
"rtc_config": null,
|
| 194 |
-
"compile_model":
|
| 195 |
"compile_mode": "max-autotune"
|
| 196 |
},
|
| 197 |
"reward_model": null,
|
|
@@ -205,7 +205,7 @@
|
|
| 205 |
"prefetch_factor": 4,
|
| 206 |
"persistent_workers": true,
|
| 207 |
"dataloader_multiprocessing_context": "spawn",
|
| 208 |
-
"steps":
|
| 209 |
"env_eval_freq": 20000,
|
| 210 |
"log_freq": 100,
|
| 211 |
"eval_steps": 0,
|
|
@@ -246,7 +246,7 @@
|
|
| 246 |
"project": "lerobot",
|
| 247 |
"entity": null,
|
| 248 |
"notes": null,
|
| 249 |
-
"run_id": "
|
| 250 |
"mode": null,
|
| 251 |
"add_tags": true
|
| 252 |
},
|
|
|
|
| 191 |
"min_period": 0.004,
|
| 192 |
"max_period": 4.0,
|
| 193 |
"rtc_config": null,
|
| 194 |
+
"compile_model": true,
|
| 195 |
"compile_mode": "max-autotune"
|
| 196 |
},
|
| 197 |
"reward_model": null,
|
|
|
|
| 205 |
"prefetch_factor": 4,
|
| 206 |
"persistent_workers": true,
|
| 207 |
"dataloader_multiprocessing_context": "spawn",
|
| 208 |
+
"steps": 2000,
|
| 209 |
"env_eval_freq": 20000,
|
| 210 |
"log_freq": 100,
|
| 211 |
"eval_steps": 0,
|
|
|
|
| 246 |
"project": "lerobot",
|
| 247 |
"entity": null,
|
| 248 |
"notes": null,
|
| 249 |
+
"run_id": "tiy82l7x",
|
| 250 |
"mode": null,
|
| 251 |
"add_tags": true
|
| 252 |
},
|