Upload policy weights, train config and readme
Browse files- model.safetensors +1 -1
- train_config.json +5 -40
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 906713296
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1fad25e5fba6edd713bc19e03cfd78f9c21275d0004683ffddfc5a3578276a01
|
| 3 |
size 906713296
|
train_config.json
CHANGED
|
@@ -64,42 +64,7 @@
|
|
| 64 |
"use_imagenet_stats": true,
|
| 65 |
"video_backend": "torchcodec"
|
| 66 |
},
|
| 67 |
-
"env":
|
| 68 |
-
"type": "aloha",
|
| 69 |
-
"task": "AlohaInsertion-v0",
|
| 70 |
-
"fps": 50,
|
| 71 |
-
"features": {
|
| 72 |
-
"action": {
|
| 73 |
-
"type": "ACTION",
|
| 74 |
-
"shape": [
|
| 75 |
-
14
|
| 76 |
-
]
|
| 77 |
-
},
|
| 78 |
-
"agent_pos": {
|
| 79 |
-
"type": "STATE",
|
| 80 |
-
"shape": [
|
| 81 |
-
14
|
| 82 |
-
]
|
| 83 |
-
},
|
| 84 |
-
"pixels/top": {
|
| 85 |
-
"type": "VISUAL",
|
| 86 |
-
"shape": [
|
| 87 |
-
480,
|
| 88 |
-
640,
|
| 89 |
-
3
|
| 90 |
-
]
|
| 91 |
-
}
|
| 92 |
-
},
|
| 93 |
-
"features_map": {
|
| 94 |
-
"action": "action",
|
| 95 |
-
"agent_pos": "observation.state",
|
| 96 |
-
"top": "observation.image.top",
|
| 97 |
-
"pixels/top": "observation.images.top"
|
| 98 |
-
},
|
| 99 |
-
"episode_length": 400,
|
| 100 |
-
"obs_type": "pixels_agent_pos",
|
| 101 |
-
"render_mode": "rgb_array"
|
| 102 |
-
},
|
| 103 |
"policy": {
|
| 104 |
"type": "smolvla",
|
| 105 |
"n_obs_steps": 1,
|
|
@@ -188,14 +153,14 @@
|
|
| 188 |
"min_period": 0.004,
|
| 189 |
"max_period": 4.0
|
| 190 |
},
|
| 191 |
-
"output_dir": "outputs/train/
|
| 192 |
"job_name": "my_smolvla_training",
|
| 193 |
"resume": false,
|
| 194 |
"seed": 1000,
|
| 195 |
"num_workers": 4,
|
| 196 |
-
"batch_size":
|
| 197 |
"steps": 2000,
|
| 198 |
-
"eval_freq":
|
| 199 |
"log_freq": 200,
|
| 200 |
"save_checkpoint": true,
|
| 201 |
"save_freq": 20000,
|
|
@@ -229,7 +194,7 @@
|
|
| 229 |
"project": "lerobot",
|
| 230 |
"entity": null,
|
| 231 |
"notes": null,
|
| 232 |
-
"run_id": "
|
| 233 |
"mode": null
|
| 234 |
}
|
| 235 |
}
|
|
|
|
| 64 |
"use_imagenet_stats": true,
|
| 65 |
"video_backend": "torchcodec"
|
| 66 |
},
|
| 67 |
+
"env": null,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 68 |
"policy": {
|
| 69 |
"type": "smolvla",
|
| 70 |
"n_obs_steps": 1,
|
|
|
|
| 153 |
"min_period": 0.004,
|
| 154 |
"max_period": 4.0
|
| 155 |
},
|
| 156 |
+
"output_dir": "outputs/train/my_smolvla",
|
| 157 |
"job_name": "my_smolvla_training",
|
| 158 |
"resume": false,
|
| 159 |
"seed": 1000,
|
| 160 |
"num_workers": 4,
|
| 161 |
+
"batch_size": 64,
|
| 162 |
"steps": 2000,
|
| 163 |
+
"eval_freq": 200,
|
| 164 |
"log_freq": 200,
|
| 165 |
"save_checkpoint": true,
|
| 166 |
"save_freq": 20000,
|
|
|
|
| 194 |
"project": "lerobot",
|
| 195 |
"entity": null,
|
| 196 |
"notes": null,
|
| 197 |
+
"run_id": "c1rojnq8",
|
| 198 |
"mode": null
|
| 199 |
}
|
| 200 |
}
|