Robotics
LeRobot
Safetensors
smolvla
hafeezjimoh commited on
Commit
ae081c8
·
verified ·
1 Parent(s): 72e7602

Upload policy weights, train config and readme

Browse files
Files changed (3) hide show
  1. config.json +1 -1
  2. model.safetensors +1 -1
  3. train_config.json +5 -5
config.json CHANGED
@@ -48,7 +48,7 @@
48
  "private": null,
49
  "tags": null,
50
  "license": null,
51
- "pretrained_path": "lerobot/smolvla_base",
52
  "chunk_size": 50,
53
  "n_action_steps": 50,
54
  "normalization_mapping": {
 
48
  "private": null,
49
  "tags": null,
50
  "license": null,
51
+ "pretrained_path": "outputs/train/smolvla_pickcube/checkpoints/last/pretrained_model",
52
  "chunk_size": 50,
53
  "n_action_steps": 50,
54
  "normalization_mapping": {
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6b8f8c42d15d7b0aa54732b39af61ba8454686eba6c34af0d7582dd4ff3cba58
3
  size 906712520
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fd80c0d95c6ea35542972d1a45fcc84d3a55af3c09ccac2196d64eba6b1f9656
3
  size 906712520
train_config.json CHANGED
@@ -130,7 +130,7 @@
130
  "private": null,
131
  "tags": null,
132
  "license": null,
133
- "pretrained_path": "lerobot/smolvla_base",
134
  "chunk_size": 50,
135
  "n_action_steps": 50,
136
  "normalization_mapping": {
@@ -179,15 +179,15 @@
179
  },
180
  "output_dir": "outputs/train/smolvla_pickcube",
181
  "job_name": "my_smolvla_training",
182
- "resume": false,
183
  "seed": 1000,
184
  "num_workers": 4,
185
  "batch_size": 64,
186
- "steps": 10000,
187
  "eval_freq": 20000,
188
  "log_freq": 200,
189
  "save_checkpoint": true,
190
- "save_freq": 20000,
191
  "use_policy_training_preset": true,
192
  "optimizer": {
193
  "type": "adamw",
@@ -221,6 +221,6 @@
221
  "run_id": "5ieklcy4",
222
  "mode": null
223
  },
224
- "checkpoint_path": null,
225
  "rename_map": {}
226
  }
 
130
  "private": null,
131
  "tags": null,
132
  "license": null,
133
+ "pretrained_path": "outputs/train/smolvla_pickcube/checkpoints/last/pretrained_model",
134
  "chunk_size": 50,
135
  "n_action_steps": 50,
136
  "normalization_mapping": {
 
179
  },
180
  "output_dir": "outputs/train/smolvla_pickcube",
181
  "job_name": "my_smolvla_training",
182
+ "resume": true,
183
  "seed": 1000,
184
  "num_workers": 4,
185
  "batch_size": 64,
186
+ "steps": 20000,
187
  "eval_freq": 20000,
188
  "log_freq": 200,
189
  "save_checkpoint": true,
190
+ "save_freq": 1000,
191
  "use_policy_training_preset": true,
192
  "optimizer": {
193
  "type": "adamw",
 
221
  "run_id": "5ieklcy4",
222
  "mode": null
223
  },
224
+ "checkpoint_path": "outputs/train/smolvla_pickcube/checkpoints/last",
225
  "rename_map": {}
226
  }