Robotics
LeRobot
Safetensors
smolvla
TANAY779 commited on
Commit
637fa28
·
verified ·
1 Parent(s): d5f02c1

Upload policy weights, train config and readme

Browse files
Files changed (4) hide show
  1. README.md +63 -0
  2. config.json +1 -1
  3. model.safetensors +1 -1
  4. train_config.json +3 -3
README.md ADDED
@@ -0,0 +1,63 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: lerobot/smolvla_base
3
+ datasets: TANAY779/SmolVLA_Stacking
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
+
14
+ # Model Card for smolvla
15
+
16
+ <!-- Provide a quick summary of what the model is/does. -->
17
+
18
+
19
+ [SmolVLA](https://huggingface.co/papers/2506.01844) is a compact, efficient vision-language-action model that achieves competitive performance at reduced computational costs and can be deployed on consumer-grade hardware.
20
+
21
+
22
+ This policy has been trained and pushed to the Hub using [LeRobot](https://github.com/huggingface/lerobot).
23
+ See the full documentation at [LeRobot Docs](https://huggingface.co/docs/lerobot/index).
24
+
25
+ ---
26
+
27
+ ## How to Get Started with the Model
28
+
29
+ For a complete walkthrough, see the [training guide](https://huggingface.co/docs/lerobot/il_robots#train-a-policy).
30
+ Below is the short version on how to train and run inference/eval:
31
+
32
+ ### Train from scratch
33
+
34
+ ```bash
35
+ lerobot-train \
36
+ --dataset.repo_id=${HF_USER}/<dataset> \
37
+ --policy.type=act \
38
+ --output_dir=outputs/train/<desired_policy_repo_id> \
39
+ --job_name=lerobot_training \
40
+ --policy.device=cuda \
41
+ --policy.repo_id=${HF_USER}/<desired_policy_repo_id>
42
+ --wandb.enable=true
43
+ ```
44
+
45
+ _Writes checkpoints to `outputs/train/<desired_policy_repo_id>/checkpoints/`._
46
+
47
+ ### Evaluate the policy/run inference
48
+
49
+ ```bash
50
+ lerobot-record \
51
+ --robot.type=so100_follower \
52
+ --dataset.repo_id=<hf_user>/eval_<dataset> \
53
+ --policy.path=<hf_user>/<desired_policy_repo_id> \
54
+ --episodes=10
55
+ ```
56
+
57
+ Prefix the dataset repo with **eval\_** and supply `--policy.path` pointing to a local or hub checkpoint.
58
+
59
+ ---
60
+
61
+ ## Model Details
62
+
63
+ - **License:** apache-2.0
config.json CHANGED
@@ -57,7 +57,7 @@
57
  "private": null,
58
  "tags": null,
59
  "license": null,
60
- "pretrained_path": "lerobot/smolvla_base",
61
  "chunk_size": 50,
62
  "n_action_steps": 50,
63
  "normalization_mapping": {
 
57
  "private": null,
58
  "tags": null,
59
  "license": null,
60
+ "pretrained_path": "outputs/train/SmolVLA_Stacking_3_Cubes/checkpoints/012000/pretrained_model",
61
  "chunk_size": 50,
62
  "n_action_steps": 50,
63
  "normalization_mapping": {
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a079b0648960250f64a33231c68a8640dbbe06b99e89ff48f22c05985eea6042
3
  size 906712520
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6d6b4270348b2c2640d0220302a28277defd2475879f2aa12057a143c96bd8c1
3
  size 906712520
train_config.json CHANGED
@@ -140,7 +140,7 @@
140
  "private": null,
141
  "tags": null,
142
  "license": null,
143
- "pretrained_path": "lerobot/smolvla_base",
144
  "chunk_size": 50,
145
  "n_action_steps": 50,
146
  "normalization_mapping": {
@@ -192,7 +192,7 @@
192
  },
193
  "output_dir": "outputs/train/SmolVLA_Stacking_3_Cubes",
194
  "job_name": "SmolVLA_Stacking_3_Cubes",
195
- "resume": false,
196
  "seed": 1000,
197
  "cudnn_deterministic": false,
198
  "num_workers": 4,
@@ -249,5 +249,5 @@
249
  "observation.images.overhead": "observation.images.camera1",
250
  "observation.images.wrist": "observation.images.camera2"
251
  },
252
- "checkpoint_path": null
253
  }
 
140
  "private": null,
141
  "tags": null,
142
  "license": null,
143
+ "pretrained_path": "outputs/train/SmolVLA_Stacking_3_Cubes/checkpoints/012000/pretrained_model",
144
  "chunk_size": 50,
145
  "n_action_steps": 50,
146
  "normalization_mapping": {
 
192
  },
193
  "output_dir": "outputs/train/SmolVLA_Stacking_3_Cubes",
194
  "job_name": "SmolVLA_Stacking_3_Cubes",
195
+ "resume": true,
196
  "seed": 1000,
197
  "cudnn_deterministic": false,
198
  "num_workers": 4,
 
249
  "observation.images.overhead": "observation.images.camera1",
250
  "observation.images.wrist": "observation.images.camera2"
251
  },
252
+ "checkpoint_path": "outputs/train/SmolVLA_Stacking_3_Cubes/checkpoints/012000"
253
  }