Hugging Face's logo

felsager
/
smolvla_ms_cube_stack

Robotics
LeRobot
Safetensors
smolvla
Model card Files
xet
 Community
felsager commited on
Commit
e2c779f
·
verified ·
1 Parent(s): 428cf73

Upload policy weights, train config and readme

Browse files
Files changed (4) hide show
  1. README.md +4 -4
  2. config.json +1 -1
  3. model.safetensors +1 -1
  4. train_config.json +12 -9
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
@@ -86,11 +86,11 @@ The policy consumes these observation features and produces these action feature
86
 
87
  | Setting | Value |
88
  | --- | --- |
89
- | Training steps | 2000 |
90
- | Batch size | 224 |
91
  | Optimizer | adamw |
92
  | Learning rate | 0.0001 |
93
- | Seed | 1000 |
94
  | LeRobot version | 0.6.1 |
95
 
96
  ---
 
7
  pipeline_tag: robotics
8
  tags:
9
  - lerobot
 
10
  - robotics
11
+ - smolvla
12
  ---
13
 
14
  # Model Card for smolvla
 
86
 
87
  | Setting | Value |
88
  | --- | --- |
89
+ | Training steps | 100 |
90
+ | Batch size | 256 |
91
  | Optimizer | adamw |
92
  | Learning rate | 0.0001 |
93
+ | Seed | 0 |
94
  | LeRobot version | 0.6.1 |
95
 
96
  ---
config.json CHANGED
@@ -106,5 +106,5 @@
106
  "max_period": 4.0,
107
  "rtc_config": null,
108
  "compile_model": true,
109
- "compile_mode": "max-autotune"
110
  }
 
106
  "max_period": 4.0,
107
  "rtc_config": null,
108
  "compile_model": true,
109
+ "compile_mode": "default"
110
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:34a32220eb66320e357aac989553a7b979baecefb5565f34fd700b68a419208e
3
  size 906712520
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:42525dcdf136e398721f7046ebebc9e09739423c7e2624dff3603d99e7d88366
3
  size 906712520
train_config.json CHANGED
@@ -3,7 +3,10 @@
3
  "repo_id": "felsager/ManiSkill_StackCube-v1_mp",
4
  "repo_type": "dataset",
5
  "root": null,
6
- "episodes": null,
 
 
 
7
  "image_transforms": {
8
  "enable": false,
9
  "max_num_transforms": 3,
@@ -77,7 +80,7 @@
77
  },
78
  "revision": "state-joints-only",
79
  "use_imagenet_stats": true,
80
- "video_backend": "pyav",
81
  "return_uint8": false,
82
  "depth_output_unit": "mm",
83
  "streaming": false,
@@ -192,22 +195,22 @@
192
  "max_period": 4.0,
193
  "rtc_config": null,
194
  "compile_model": true,
195
- "compile_mode": "max-autotune"
196
  },
197
  "reward_model": null,
198
- "output_dir": "/work/sim_vla_ws/runs/train/smolvla_maniskill_cube_stacking",
199
  "job_name": "smolvla_training",
200
  "resume": false,
201
- "seed": 1000,
202
  "cudnn_deterministic": false,
203
  "num_workers": 4,
204
- "batch_size": 224,
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,
212
  "max_eval_samples": 0,
213
  "tolerance_s": 0.0001,
@@ -246,7 +249,7 @@
246
  "project": "lerobot",
247
  "entity": null,
248
  "notes": null,
249
- "run_id": "tiy82l7x",
250
  "mode": null,
251
  "add_tags": true
252
  },
 
3
  "repo_id": "felsager/ManiSkill_StackCube-v1_mp",
4
  "repo_type": "dataset",
5
  "root": null,
6
+ "episodes": [
7
+ 0,
8
+ 1
9
+ ],
10
  "image_transforms": {
11
  "enable": false,
12
  "max_num_transforms": 3,
 
80
  },
81
  "revision": "state-joints-only",
82
  "use_imagenet_stats": true,
83
+ "video_backend": "torchcodec",
84
  "return_uint8": false,
85
  "depth_output_unit": "mm",
86
  "streaming": false,
 
195
  "max_period": 4.0,
196
  "rtc_config": null,
197
  "compile_model": true,
198
+ "compile_mode": "default"
199
  },
200
  "reward_model": null,
201
+ "output_dir": "/work/sim_vla_ws/runs/train/smolvla_maniskill_cube_stacking_test_3",
202
  "job_name": "smolvla_training",
203
  "resume": false,
204
+ "seed": 0,
205
  "cudnn_deterministic": false,
206
  "num_workers": 4,
207
+ "batch_size": 256,
208
  "prefetch_factor": 4,
209
  "persistent_workers": true,
210
  "dataloader_multiprocessing_context": "spawn",
211
+ "steps": 100,
212
  "env_eval_freq": 20000,
213
+ "log_freq": 10,
214
  "eval_steps": 0,
215
  "max_eval_samples": 0,
216
  "tolerance_s": 0.0001,
 
249
  "project": "lerobot",
250
  "entity": null,
251
  "notes": null,
252
+ "run_id": "6llz9c20",
253
  "mode": null,
254
  "add_tags": true
255
  },