Robotics
LeRobot
Safetensors
smolvla
pepijn223 HF Staff commited on
Commit
0166b92
·
verified ·
1 Parent(s): 2100f1b

Upload policy weights, train config and readme

Browse files
Files changed (4) hide show
  1. README.md +1 -1
  2. config.json +2 -2
  3. model.safetensors +2 -2
  4. 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": 2,
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": false,
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:3d766fcce6c4358cf746431260686c502c6c952ca6da840fd020b30e480fdd94
3
- size 422756192
 
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/nano_vla_special1",
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": false,
172
  "num_workers": 4,
173
- "batch_size": 2,
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": "7i22ennf",
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
  }