Robotics
LeRobot
Safetensors
smolvla
vpraise00 commited on
Commit
b4e4550
·
verified ·
1 Parent(s): a10b422

Upload policy weights, train config and readme

Browse files
Files changed (4) hide show
  1. README.md +1 -1
  2. config.json +0 -16
  3. model.safetensors +1 -1
  4. train_config.json +17 -29
README.md CHANGED
@@ -7,8 +7,8 @@ model_name: smolvla
7
  pipeline_tag: robotics
8
  tags:
9
  - robotics
10
- - smolvla
11
  - lerobot
 
12
  ---
13
 
14
  # Model Card for smolvla
 
7
  pipeline_tag: robotics
8
  tags:
9
  - robotics
 
10
  - lerobot
11
+ - smolvla
12
  ---
13
 
14
  # Model Card for smolvla
config.json CHANGED
@@ -31,22 +31,6 @@
31
  256,
32
  256
33
  ]
34
- },
35
- "observation.images.top": {
36
- "type": "VISUAL",
37
- "shape": [
38
- 3,
39
- 480,
40
- 640
41
- ]
42
- },
43
- "observation.images.left_wrist": {
44
- "type": "VISUAL",
45
- "shape": [
46
- 3,
47
- 480,
48
- 640
49
- ]
50
  }
51
  },
52
  "output_features": {
 
31
  256,
32
  256
33
  ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
34
  }
35
  },
36
  "output_features": {
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c6d92e801b328682012b4796af11a859e0fa24fb8e01a161e74f3addf8a9b270
3
  size 906712520
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:73e71e64e05631b4ffe51343e02571d11a61da383d0b36e1d5068e77b9377529
3
  size 906712520
train_config.json CHANGED
@@ -76,7 +76,7 @@
76
  },
77
  "revision": "main",
78
  "use_imagenet_stats": true,
79
- "video_backend": "torchcodec",
80
  "streaming": false
81
  },
82
  "env": null,
@@ -113,22 +113,6 @@
113
  256,
114
  256
115
  ]
116
- },
117
- "observation.images.top": {
118
- "type": "VISUAL",
119
- "shape": [
120
- 3,
121
- 480,
122
- 640
123
- ]
124
- },
125
- "observation.images.left_wrist": {
126
- "type": "VISUAL",
127
- "shape": [
128
- 3,
129
- 480,
130
- 640
131
- ]
132
  }
133
  },
134
  "output_features": {
@@ -197,24 +181,29 @@
197
  "compile_model": false,
198
  "compile_mode": "max-autotune"
199
  },
200
- "output_dir": "/workspace/train_with_lerobot/outputs/train/smolVLA_table7_harmful_id_pnp_100epi_50ep_20260528_210631",
201
- "job_name": "smolVLA_table7_harmful_id_pnp_100epi_50ep_20260528_210631",
202
  "resume": false,
203
  "seed": 1000,
204
  "cudnn_deterministic": false,
205
  "num_workers": 16,
206
- "persistent_workers": true,
 
 
207
  "batch_size": 128,
208
  "gradient_accumulation_steps": 1,
 
 
 
 
209
  "steps": 6513,
210
  "eval_freq": 0,
211
  "log_freq": 100,
212
- "tolerance_s": 0.0001,
213
  "save_checkpoint": true,
214
  "save_freq": 6513,
215
- "checkpoint_push_to_hub": false,
216
- "checkpoint_hub_repo_id": null,
217
- "checkpoint_hub_private": true,
218
  "use_policy_training_preset": true,
219
  "optimizer": {
220
  "type": "adamw",
@@ -236,8 +225,8 @@
236
  },
237
  "eval": {
238
  "n_episodes": 50,
239
- "batch_size": 44,
240
- "use_async_envs": true
241
  },
242
  "wandb": {
243
  "enable": true,
@@ -245,9 +234,8 @@
245
  "project": "lerobot-real-smolvla",
246
  "entity": null,
247
  "notes": null,
248
- "run_id": "qr5rcyhb",
249
- "mode": "online",
250
- "add_tags": true
251
  },
252
  "peft": null,
253
  "use_rabc": false,
 
76
  },
77
  "revision": "main",
78
  "use_imagenet_stats": true,
79
+ "video_backend": "pyav",
80
  "streaming": false
81
  },
82
  "env": null,
 
113
  256,
114
  256
115
  ]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
116
  }
117
  },
118
  "output_features": {
 
181
  "compile_model": false,
182
  "compile_mode": "max-autotune"
183
  },
184
+ "output_dir": "/home/work/hscho/corl_2026/AutoDataCollector/lerobot/outputs/train/smolVLA_table7_harmful_id_pnp_100epi_50ep_20260528_222310",
185
+ "job_name": "smolVLA_table7_harmful_id_pnp_100epi_50ep_20260528_222310",
186
  "resume": false,
187
  "seed": 1000,
188
  "cudnn_deterministic": false,
189
  "num_workers": 16,
190
+ "dataloader_prefetch_factor": 2,
191
+ "dataloader_persistent_workers": true,
192
+ "dataloader_pin_memory": true,
193
  "batch_size": 128,
194
  "gradient_accumulation_steps": 1,
195
+ "ddp_find_unused_parameters": true,
196
+ "profile_timing": false,
197
+ "preprocess_in_workers": false,
198
+ "groot_preprocess_in_workers": false,
199
  "steps": 6513,
200
  "eval_freq": 0,
201
  "log_freq": 100,
202
+ "tolerance_s": 0.04,
203
  "save_checkpoint": true,
204
  "save_freq": 6513,
205
+ "push_checkpoint_to_hub": false,
206
+ "checkpoint_repo_id": null,
 
207
  "use_policy_training_preset": true,
208
  "optimizer": {
209
  "type": "adamw",
 
225
  },
226
  "eval": {
227
  "n_episodes": 50,
228
+ "batch_size": 50,
229
+ "use_async_envs": false
230
  },
231
  "wandb": {
232
  "enable": true,
 
234
  "project": "lerobot-real-smolvla",
235
  "entity": null,
236
  "notes": null,
237
+ "run_id": "tdlhw428",
238
+ "mode": "online"
 
239
  },
240
  "peft": null,
241
  "use_rabc": false,