Upload policy weights, train config and readme
Browse files- README.md +2 -2
- config.json +2 -2
- model.safetensors +1 -1
- train_config.json +7 -10
README.md
CHANGED
|
@@ -1,13 +1,13 @@
|
|
| 1 |
---
|
| 2 |
base_model: lerobot/smolvla_base
|
| 3 |
-
datasets: PRFitz/lekiwi-dataset-test-
|
| 4 |
library_name: lerobot
|
| 5 |
license: apache-2.0
|
| 6 |
model_name: smolvla
|
| 7 |
pipeline_tag: robotics
|
| 8 |
tags:
|
| 9 |
-
- lerobot
|
| 10 |
- robotics
|
|
|
|
| 11 |
- smolvla
|
| 12 |
---
|
| 13 |
|
|
|
|
| 1 |
---
|
| 2 |
base_model: lerobot/smolvla_base
|
| 3 |
+
datasets: PRFitz/lekiwi-dataset-test-left
|
| 4 |
library_name: lerobot
|
| 5 |
license: apache-2.0
|
| 6 |
model_name: smolvla
|
| 7 |
pipeline_tag: robotics
|
| 8 |
tags:
|
|
|
|
| 9 |
- robotics
|
| 10 |
+
- lerobot
|
| 11 |
- smolvla
|
| 12 |
---
|
| 13 |
|
config.json
CHANGED
|
@@ -40,7 +40,7 @@
|
|
| 40 |
"private": null,
|
| 41 |
"tags": null,
|
| 42 |
"license": null,
|
| 43 |
-
"pretrained_path":
|
| 44 |
"chunk_size": 50,
|
| 45 |
"n_action_steps": 50,
|
| 46 |
"normalization_mapping": {
|
|
@@ -70,7 +70,7 @@
|
|
| 70 |
],
|
| 71 |
"optimizer_eps": 1e-08,
|
| 72 |
"optimizer_weight_decay": 1e-10,
|
| 73 |
-
"optimizer_grad_clip_norm": 10,
|
| 74 |
"scheduler_warmup_steps": 1000,
|
| 75 |
"scheduler_decay_steps": 30000,
|
| 76 |
"scheduler_decay_lr": 2.5e-06,
|
|
|
|
| 40 |
"private": null,
|
| 41 |
"tags": null,
|
| 42 |
"license": null,
|
| 43 |
+
"pretrained_path": "PRFitz/circle9_smolvla",
|
| 44 |
"chunk_size": 50,
|
| 45 |
"n_action_steps": 50,
|
| 46 |
"normalization_mapping": {
|
|
|
|
| 70 |
],
|
| 71 |
"optimizer_eps": 1e-08,
|
| 72 |
"optimizer_weight_decay": 1e-10,
|
| 73 |
+
"optimizer_grad_clip_norm": 10.0,
|
| 74 |
"scheduler_warmup_steps": 1000,
|
| 75 |
"scheduler_decay_steps": 30000,
|
| 76 |
"scheduler_decay_lr": 2.5e-06,
|
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:f7c4fe53564c5c2c866264230bfadae494818b4682f07c14d12610951aa26f64
|
| 3 |
size 906712520
|
train_config.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"dataset": {
|
| 3 |
-
"repo_id": "PRFitz/lekiwi-dataset-test-
|
| 4 |
"root": null,
|
| 5 |
"episodes": null,
|
| 6 |
"image_transforms": {
|
|
@@ -122,7 +122,7 @@
|
|
| 122 |
"private": null,
|
| 123 |
"tags": null,
|
| 124 |
"license": null,
|
| 125 |
-
"pretrained_path":
|
| 126 |
"chunk_size": 50,
|
| 127 |
"n_action_steps": 50,
|
| 128 |
"normalization_mapping": {
|
|
@@ -152,7 +152,7 @@
|
|
| 152 |
],
|
| 153 |
"optimizer_eps": 1e-08,
|
| 154 |
"optimizer_weight_decay": 1e-10,
|
| 155 |
-
"optimizer_grad_clip_norm": 10,
|
| 156 |
"scheduler_warmup_steps": 1000,
|
| 157 |
"scheduler_decay_steps": 30000,
|
| 158 |
"scheduler_decay_lr": 2.5e-06,
|
|
@@ -169,8 +169,8 @@
|
|
| 169 |
"min_period": 0.004,
|
| 170 |
"max_period": 4.0
|
| 171 |
},
|
| 172 |
-
"output_dir": "outputs/train/
|
| 173 |
-
"job_name": "
|
| 174 |
"resume": false,
|
| 175 |
"seed": 1000,
|
| 176 |
"num_workers": 4,
|
|
@@ -185,7 +185,7 @@
|
|
| 185 |
"type": "adamw",
|
| 186 |
"lr": 0.0001,
|
| 187 |
"weight_decay": 1e-10,
|
| 188 |
-
"grad_clip_norm": 10,
|
| 189 |
"betas": [
|
| 190 |
0.9,
|
| 191 |
0.95
|
|
@@ -214,8 +214,5 @@
|
|
| 214 |
"mode": null
|
| 215 |
},
|
| 216 |
"checkpoint_path": null,
|
| 217 |
-
"rename_map": {
|
| 218 |
-
"observation.images.wrist": "observation.images.camera1",
|
| 219 |
-
"observation.images.front": "observation.images.camera2"
|
| 220 |
-
}
|
| 221 |
}
|
|
|
|
| 1 |
{
|
| 2 |
"dataset": {
|
| 3 |
+
"repo_id": "PRFitz/lekiwi-dataset-test-left",
|
| 4 |
"root": null,
|
| 5 |
"episodes": null,
|
| 6 |
"image_transforms": {
|
|
|
|
| 122 |
"private": null,
|
| 123 |
"tags": null,
|
| 124 |
"license": null,
|
| 125 |
+
"pretrained_path": "PRFitz/circle9_smolvla",
|
| 126 |
"chunk_size": 50,
|
| 127 |
"n_action_steps": 50,
|
| 128 |
"normalization_mapping": {
|
|
|
|
| 152 |
],
|
| 153 |
"optimizer_eps": 1e-08,
|
| 154 |
"optimizer_weight_decay": 1e-10,
|
| 155 |
+
"optimizer_grad_clip_norm": 10.0,
|
| 156 |
"scheduler_warmup_steps": 1000,
|
| 157 |
"scheduler_decay_steps": 30000,
|
| 158 |
"scheduler_decay_lr": 2.5e-06,
|
|
|
|
| 169 |
"min_period": 0.004,
|
| 170 |
"max_period": 4.0
|
| 171 |
},
|
| 172 |
+
"output_dir": "outputs/train/circle9_smolvla2",
|
| 173 |
+
"job_name": "smolvla_2",
|
| 174 |
"resume": false,
|
| 175 |
"seed": 1000,
|
| 176 |
"num_workers": 4,
|
|
|
|
| 185 |
"type": "adamw",
|
| 186 |
"lr": 0.0001,
|
| 187 |
"weight_decay": 1e-10,
|
| 188 |
+
"grad_clip_norm": 10.0,
|
| 189 |
"betas": [
|
| 190 |
0.9,
|
| 191 |
0.95
|
|
|
|
| 214 |
"mode": null
|
| 215 |
},
|
| 216 |
"checkpoint_path": null,
|
| 217 |
+
"rename_map": {}
|
|
|
|
|
|
|
|
|
|
| 218 |
}
|