Robotics
LeRobot
Safetensors
smolvla
Beable commited on
Commit
69a0fcf
·
verified ·
1 Parent(s): 3563a8b

Upload 3 files

Browse files
Files changed (3) hide show
  1. config.json +3 -3
  2. model.safetensors +1 -1
  3. train_config.json +13 -13
config.json CHANGED
@@ -11,8 +11,8 @@
11
  "type": "VISUAL",
12
  "shape": [
13
  3,
14
- 720,
15
- 1280
16
  ]
17
  },
18
  "observation.state": {
@@ -54,7 +54,7 @@
54
  "freeze_vision_encoder": false,
55
  "train_expert_only": false,
56
  "train_state_proj": true,
57
- "optimizer_lr": 1e-05,
58
  "optimizer_betas": [
59
  0.9,
60
  0.95
 
11
  "type": "VISUAL",
12
  "shape": [
13
  3,
14
+ 360,
15
+ 640
16
  ]
17
  },
18
  "observation.state": {
 
54
  "freeze_vision_encoder": false,
55
  "train_expert_only": false,
56
  "train_state_proj": true,
57
+ "optimizer_lr": 8e-06,
58
  "optimizer_betas": [
59
  0.9,
60
  0.95
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:41f9cc12e3298863908f85a0dbf8696f88a6f9f8ed256ebbfbd594de9351197a
3
  size 1197790016
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5d35760fe5e5929331d8374a643102ca50ae054858b699b6657a664e26106ad3
3
  size 1197790016
train_config.json CHANGED
@@ -1,12 +1,12 @@
1
  {
2
  "dataset": {
3
- "repo_id": "Beable/lerobot-SOARM100-sim2",
4
  "root": null,
5
  "episodes": null,
6
  "image_transforms": {
7
  "enable": true,
8
- "max_num_transforms": 3,
9
- "random_order": false,
10
  "tfs": {
11
  "brightness": {
12
  "weight": 1.0,
@@ -60,10 +60,10 @@
60
  }
61
  }
62
  },
63
- "revision": "v2.1",
64
  "use_imagenet_stats": true,
65
  "video_backend": "torchcodec",
66
- "tolerance_s": 0.02,
67
  "timestamps_check": "warn"
68
  },
69
  "env": null,
@@ -80,8 +80,8 @@
80
  "type": "VISUAL",
81
  "shape": [
82
  3,
83
- 720,
84
- 1280
85
  ]
86
  },
87
  "observation.state": {
@@ -123,7 +123,7 @@
123
  "freeze_vision_encoder": false,
124
  "train_expert_only": false,
125
  "train_state_proj": true,
126
- "optimizer_lr": 1e-05,
127
  "optimizer_betas": [
128
  0.9,
129
  0.95
@@ -147,21 +147,21 @@
147
  "min_period": 0.004,
148
  "max_period": 4.0
149
  },
150
- "output_dir": "outputs/train/2025-08-14/02-35-56_smolvla",
151
  "job_name": "smolvla",
152
  "resume": false,
153
  "seed": 1000,
154
  "num_workers": 4,
155
  "batch_size": 4,
156
- "steps": 15000,
157
  "eval_freq": 20000,
158
  "log_freq": 100,
159
  "save_checkpoint": true,
160
- "save_freq": 2000,
161
  "use_policy_training_preset": true,
162
  "optimizer": {
163
  "type": "adamw",
164
- "lr": 1e-05,
165
  "weight_decay": 0.0001,
166
  "grad_clip_norm": 10,
167
  "betas": [
@@ -174,7 +174,7 @@
174
  "type": "cosine_decay_with_warmup",
175
  "num_warmup_steps": 1000,
176
  "num_decay_steps": 30000,
177
- "peak_lr": 1e-05,
178
  "decay_lr": 2.5e-06
179
  },
180
  "eval": {
 
1
  {
2
  "dataset": {
3
+ "repo_id": "Beable/lerobot-SOARM100-sim3",
4
  "root": null,
5
  "episodes": null,
6
  "image_transforms": {
7
  "enable": true,
8
+ "max_num_transforms": 4,
9
+ "random_order": true,
10
  "tfs": {
11
  "brightness": {
12
  "weight": 1.0,
 
60
  }
61
  }
62
  },
63
+ "revision": null,
64
  "use_imagenet_stats": true,
65
  "video_backend": "torchcodec",
66
+ "tolerance_s": 0.0001,
67
  "timestamps_check": "warn"
68
  },
69
  "env": null,
 
80
  "type": "VISUAL",
81
  "shape": [
82
  3,
83
+ 360,
84
+ 640
85
  ]
86
  },
87
  "observation.state": {
 
123
  "freeze_vision_encoder": false,
124
  "train_expert_only": false,
125
  "train_state_proj": true,
126
+ "optimizer_lr": 8e-06,
127
  "optimizer_betas": [
128
  0.9,
129
  0.95
 
147
  "min_period": 0.004,
148
  "max_period": 4.0
149
  },
150
+ "output_dir": "outputs/train/2025-08-14/18-38-08_smolvla",
151
  "job_name": "smolvla",
152
  "resume": false,
153
  "seed": 1000,
154
  "num_workers": 4,
155
  "batch_size": 4,
156
+ "steps": 50000,
157
  "eval_freq": 20000,
158
  "log_freq": 100,
159
  "save_checkpoint": true,
160
+ "save_freq": 30000,
161
  "use_policy_training_preset": true,
162
  "optimizer": {
163
  "type": "adamw",
164
+ "lr": 8e-06,
165
  "weight_decay": 0.0001,
166
  "grad_clip_norm": 10,
167
  "betas": [
 
174
  "type": "cosine_decay_with_warmup",
175
  "num_warmup_steps": 1000,
176
  "num_decay_steps": 30000,
177
+ "peak_lr": 8e-06,
178
  "decay_lr": 2.5e-06
179
  },
180
  "eval": {