Upload policy weights, train config and readme
Browse files- README.md +1 -1
- config.json +2 -2
- model.safetensors +1 -1
- train_config.json +8 -8
README.md
CHANGED
|
@@ -6,9 +6,9 @@ license: apache-2.0
|
|
| 6 |
model_name: smolvla
|
| 7 |
pipeline_tag: robotics
|
| 8 |
tags:
|
| 9 |
-
- lerobot
|
| 10 |
- robotics
|
| 11 |
- smolvla
|
|
|
|
| 12 |
---
|
| 13 |
|
| 14 |
# Model Card for smolvla
|
|
|
|
| 6 |
model_name: smolvla
|
| 7 |
pipeline_tag: robotics
|
| 8 |
tags:
|
|
|
|
| 9 |
- robotics
|
| 10 |
- smolvla
|
| 11 |
+
- lerobot
|
| 12 |
---
|
| 13 |
|
| 14 |
# Model Card for smolvla
|
config.json
CHANGED
|
@@ -51,8 +51,8 @@
|
|
| 51 |
"max_state_dim": 32,
|
| 52 |
"max_action_dim": 32,
|
| 53 |
"resize_imgs_with_padding": [
|
| 54 |
-
|
| 55 |
-
|
| 56 |
],
|
| 57 |
"empty_cameras": 0,
|
| 58 |
"adapt_to_pi_aloha": false,
|
|
|
|
| 51 |
"max_state_dim": 32,
|
| 52 |
"max_action_dim": 32,
|
| 53 |
"resize_imgs_with_padding": [
|
| 54 |
+
256,
|
| 55 |
+
256
|
| 56 |
],
|
| 57 |
"empty_cameras": 0,
|
| 58 |
"adapt_to_pi_aloha": false,
|
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:879c267843913973f814372e0089a65d8b27d465564ad060f86612524b428d57
|
| 3 |
size 906712520
|
train_config.json
CHANGED
|
@@ -4,7 +4,7 @@
|
|
| 4 |
"root": null,
|
| 5 |
"episodes": null,
|
| 6 |
"image_transforms": {
|
| 7 |
-
"enable":
|
| 8 |
"max_num_transforms": 3,
|
| 9 |
"random_order": false,
|
| 10 |
"tfs": {
|
|
@@ -227,8 +227,8 @@
|
|
| 227 |
"max_state_dim": 32,
|
| 228 |
"max_action_dim": 32,
|
| 229 |
"resize_imgs_with_padding": [
|
| 230 |
-
|
| 231 |
-
|
| 232 |
],
|
| 233 |
"empty_cameras": 0,
|
| 234 |
"adapt_to_pi_aloha": false,
|
|
@@ -264,14 +264,14 @@
|
|
| 264 |
"max_period": 4.0,
|
| 265 |
"rtc_config": null
|
| 266 |
},
|
| 267 |
-
"output_dir": "
|
| 268 |
"job_name": "libero_smolvla",
|
| 269 |
"resume": false,
|
| 270 |
"seed": 1000,
|
| 271 |
"num_workers": 4,
|
| 272 |
"batch_size": 4,
|
| 273 |
"steps": 100000,
|
| 274 |
-
"eval_freq":
|
| 275 |
"log_freq": 200,
|
| 276 |
"tolerance_s": 0.0001,
|
| 277 |
"save_checkpoint": true,
|
|
@@ -301,12 +301,12 @@
|
|
| 301 |
"use_async_envs": false
|
| 302 |
},
|
| 303 |
"wandb": {
|
| 304 |
-
"enable":
|
| 305 |
"disable_artifact": false,
|
| 306 |
-
"project": "
|
| 307 |
"entity": null,
|
| 308 |
"notes": null,
|
| 309 |
-
"run_id":
|
| 310 |
"mode": null
|
| 311 |
},
|
| 312 |
"use_rabc": false,
|
|
|
|
| 4 |
"root": null,
|
| 5 |
"episodes": null,
|
| 6 |
"image_transforms": {
|
| 7 |
+
"enable": true,
|
| 8 |
"max_num_transforms": 3,
|
| 9 |
"random_order": false,
|
| 10 |
"tfs": {
|
|
|
|
| 227 |
"max_state_dim": 32,
|
| 228 |
"max_action_dim": 32,
|
| 229 |
"resize_imgs_with_padding": [
|
| 230 |
+
256,
|
| 231 |
+
256
|
| 232 |
],
|
| 233 |
"empty_cameras": 0,
|
| 234 |
"adapt_to_pi_aloha": false,
|
|
|
|
| 264 |
"max_period": 4.0,
|
| 265 |
"rtc_config": null
|
| 266 |
},
|
| 267 |
+
"output_dir": "resize_outputs",
|
| 268 |
"job_name": "libero_smolvla",
|
| 269 |
"resume": false,
|
| 270 |
"seed": 1000,
|
| 271 |
"num_workers": 4,
|
| 272 |
"batch_size": 4,
|
| 273 |
"steps": 100000,
|
| 274 |
+
"eval_freq": 10000,
|
| 275 |
"log_freq": 200,
|
| 276 |
"tolerance_s": 0.0001,
|
| 277 |
"save_checkpoint": true,
|
|
|
|
| 301 |
"use_async_envs": false
|
| 302 |
},
|
| 303 |
"wandb": {
|
| 304 |
+
"enable": true,
|
| 305 |
"disable_artifact": false,
|
| 306 |
+
"project": "SmolVLA",
|
| 307 |
"entity": null,
|
| 308 |
"notes": null,
|
| 309 |
+
"run_id": "r7tuxubo",
|
| 310 |
"mode": null
|
| 311 |
},
|
| 312 |
"use_rabc": false,
|