Upload policy weights, train config and readme
Browse files- config.json +1 -1
- model.safetensors +1 -1
- train_config.json +5 -5
config.json
CHANGED
|
@@ -48,7 +48,7 @@
|
|
| 48 |
"private": null,
|
| 49 |
"tags": null,
|
| 50 |
"license": null,
|
| 51 |
-
"pretrained_path": "
|
| 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:
|
| 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": "
|
| 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":
|
| 183 |
"seed": 1000,
|
| 184 |
"num_workers": 4,
|
| 185 |
"batch_size": 64,
|
| 186 |
-
"steps":
|
| 187 |
"eval_freq": 20000,
|
| 188 |
"log_freq": 200,
|
| 189 |
"save_checkpoint": true,
|
| 190 |
-
"save_freq":
|
| 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":
|
| 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 |
}
|