Robotics
LeRobot
Safetensors
smolvla
Beable commited on
Commit
e4e36c2
·
verified ·
1 Parent(s): 20de569

Upload policy weights, train config and readme

Browse files
Files changed (4) hide show
  1. README.md +1 -1
  2. config.json +1 -1
  3. model.safetensors +1 -1
  4. train_config.json +8 -8
README.md CHANGED
@@ -1,6 +1,6 @@
1
  ---
2
  base_model: lerobot/smolvla_base
3
- datasets: Beable/lerobot-SOARM100-sim2
4
  library_name: lerobot
5
  license: apache-2.0
6
  model_name: smolvla
 
1
  ---
2
  base_model: lerobot/smolvla_base
3
+ datasets: Beable/lerobot-SOARM100-sim3
4
  library_name: lerobot
5
  license: apache-2.0
6
  model_name: smolvla
config.json CHANGED
@@ -51,7 +51,7 @@
51
  "tokenizer_max_length": 48,
52
  "num_steps": 10,
53
  "use_cache": true,
54
- "freeze_vision_encoder": true,
55
  "train_expert_only": false,
56
  "train_state_proj": true,
57
  "optimizer_lr": 2e-05,
 
51
  "tokenizer_max_length": 48,
52
  "num_steps": 10,
53
  "use_cache": true,
54
+ "freeze_vision_encoder": false,
55
  "train_expert_only": false,
56
  "train_state_proj": true,
57
  "optimizer_lr": 2e-05,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:33b74bfc1e65b4bd1c106aa84ef0b5b422e4a55a4d12287e2d1d2115068759f3
3
  size 1197790016
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3e46c3c1de6ef85d549bff9674db9178b62ca1a194a4b6c6830da3f9670d6dfb
3
  size 1197790016
train_config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "dataset": {
3
- "repo_id": "Beable/lerobot-SOARM100-sim",
4
  "root": null,
5
  "episodes": null,
6
  "image_transforms": {
@@ -60,10 +60,10 @@
60
  }
61
  }
62
  },
63
- "revision": "v2.2",
64
  "use_imagenet_stats": true,
65
  "video_backend": "torchcodec",
66
- "tolerance_s": 0.02,
67
  "timestamps_check": "warn"
68
  },
69
  "env": null,
@@ -120,7 +120,7 @@
120
  "tokenizer_max_length": 48,
121
  "num_steps": 10,
122
  "use_cache": true,
123
- "freeze_vision_encoder": true,
124
  "train_expert_only": false,
125
  "train_state_proj": true,
126
  "optimizer_lr": 2e-05,
@@ -147,17 +147,17 @@
147
  "min_period": 0.004,
148
  "max_period": 4.0
149
  },
150
- "output_dir": "outputs/train/2025-08-15/01-34-33_smolvla",
151
  "job_name": "smolvla",
152
  "resume": false,
153
  "seed": 1000,
154
  "num_workers": 4,
155
- "batch_size": 32,
156
- "steps": 2000,
157
  "eval_freq": 20000,
158
  "log_freq": 100,
159
  "save_checkpoint": true,
160
- "save_freq": 1000,
161
  "use_policy_training_preset": true,
162
  "optimizer": {
163
  "type": "adamw",
 
1
  {
2
  "dataset": {
3
+ "repo_id": "Beable/lerobot-SOARM100-sim3",
4
  "root": null,
5
  "episodes": null,
6
  "image_transforms": {
 
60
  }
61
  }
62
  },
63
+ "revision": "v2.1",
64
  "use_imagenet_stats": true,
65
  "video_backend": "torchcodec",
66
+ "tolerance_s": 0.0001,
67
  "timestamps_check": "warn"
68
  },
69
  "env": null,
 
120
  "tokenizer_max_length": 48,
121
  "num_steps": 10,
122
  "use_cache": true,
123
+ "freeze_vision_encoder": false,
124
  "train_expert_only": false,
125
  "train_state_proj": true,
126
  "optimizer_lr": 2e-05,
 
147
  "min_period": 0.004,
148
  "max_period": 4.0
149
  },
150
+ "output_dir": "outputs/train/2025-08-15/03-56-10_smolvla",
151
  "job_name": "smolvla",
152
  "resume": false,
153
  "seed": 1000,
154
  "num_workers": 4,
155
+ "batch_size": 4,
156
+ "steps": 40000,
157
  "eval_freq": 20000,
158
  "log_freq": 100,
159
  "save_checkpoint": true,
160
+ "save_freq": 40000,
161
  "use_policy_training_preset": true,
162
  "optimizer": {
163
  "type": "adamw",