Upload policy weights, train config and readme
Browse files- README.md +1 -1
- config.json +2 -2
- model.safetensors +2 -2
- train_config.json +4 -4
README.md
CHANGED
@@ -6,8 +6,8 @@ license: apache-2.0
|
|
6 |
model_name: smolvla
|
7 |
pipeline_tag: robotics
|
8 |
tags:
|
9 |
-
- lerobot
|
10 |
- smolvla
|
|
|
11 |
- robotics
|
12 |
---
|
13 |
|
|
|
6 |
model_name: smolvla
|
7 |
pipeline_tag: robotics
|
8 |
tags:
|
|
|
9 |
- smolvla
|
10 |
+
- lerobot
|
11 |
- robotics
|
12 |
---
|
13 |
|
config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"batch_size":
|
3 |
"dataset": {
|
4 |
"episodes": null,
|
5 |
"image_transforms": {
|
@@ -127,7 +127,7 @@
|
|
127 |
"vision_model_name": "google/siglip-base-patch16-512"
|
128 |
},
|
129 |
"steps": 100,
|
130 |
-
"torch_compile":
|
131 |
"wandb": {
|
132 |
"disable_artifact": false,
|
133 |
"enable": true,
|
|
|
1 |
{
|
2 |
+
"batch_size": 4,
|
3 |
"dataset": {
|
4 |
"episodes": null,
|
5 |
"image_transforms": {
|
|
|
127 |
"vision_model_name": "google/siglip-base-patch16-512"
|
128 |
},
|
129 |
"steps": 100,
|
130 |
+
"torch_compile": true,
|
131 |
"wandb": {
|
132 |
"disable_artifact": false,
|
133 |
"enable": true,
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:de43ae0693fb37b23c4c62ea7daa63dacf954a1e15636ce6b8d7d78465ca89e9
|
3 |
+
size 422768032
|
train_config.json
CHANGED
@@ -161,16 +161,16 @@
|
|
161 |
"min_period": 0.004,
|
162 |
"max_period": 4.0
|
163 |
},
|
164 |
-
"output_dir": "outputs/train/
|
165 |
"job_name": "nanovla_main",
|
166 |
"resume": false,
|
167 |
"device": "cuda",
|
168 |
"seed": 1000,
|
169 |
"nccl_timeout": 10,
|
170 |
"gradient_accumulation_steps": 1,
|
171 |
-
"torch_compile":
|
172 |
"num_workers": 4,
|
173 |
-
"batch_size":
|
174 |
"steps": 100,
|
175 |
"eval_freq": 20000,
|
176 |
"log_freq": 20,
|
@@ -206,7 +206,7 @@
|
|
206 |
"project": "lerobot",
|
207 |
"entity": null,
|
208 |
"notes": null,
|
209 |
-
"run_id": "
|
210 |
"mode": null
|
211 |
}
|
212 |
}
|
|
|
161 |
"min_period": 0.004,
|
162 |
"max_period": 4.0
|
163 |
},
|
164 |
+
"output_dir": "outputs/train/nano_vla_special2",
|
165 |
"job_name": "nanovla_main",
|
166 |
"resume": false,
|
167 |
"device": "cuda",
|
168 |
"seed": 1000,
|
169 |
"nccl_timeout": 10,
|
170 |
"gradient_accumulation_steps": 1,
|
171 |
+
"torch_compile": true,
|
172 |
"num_workers": 4,
|
173 |
+
"batch_size": 4,
|
174 |
"steps": 100,
|
175 |
"eval_freq": 20000,
|
176 |
"log_freq": 20,
|
|
|
206 |
"project": "lerobot",
|
207 |
"entity": null,
|
208 |
"notes": null,
|
209 |
+
"run_id": "gwrnt4ds",
|
210 |
"mode": null
|
211 |
}
|
212 |
}
|