Update README.md
Browse files
README.md
CHANGED
@@ -30,24 +30,11 @@ Example of finetuning the smolvla pretrained model (`smolvla_base`):
|
|
30 |
```bash
|
31 |
python lerobot/scripts/train.py \
|
32 |
--policy.path=lerobot/smolvla_base \
|
33 |
-
--dataset.repo_id=lerobot/svla_so101_pickplace \
|
34 |
-
--batch_size=64 \
|
35 |
-
--steps=20000 \
|
36 |
-
--output_dir=outputs/train/my_smolvla \
|
37 |
-
--job_name=my_smolvla_training \
|
38 |
-
--policy.device=cuda \
|
39 |
-
--wandb.enable=true
|
40 |
-
```
|
41 |
-
|
42 |
-
Example of finetuning the smolvla neural network with pretrained VLM and action expert
|
43 |
-
intialized from scratch:
|
44 |
-
```bash
|
45 |
-
python lerobot/scripts/train.py \
|
46 |
--dataset.repo_id=TekbotRobotics/svla_so101_pickplace_flags_sorting \
|
47 |
-
|
48 |
-
--steps=
|
49 |
--output_dir=outputs/train/my_smolvla \
|
50 |
--job_name=my_smolvla_training \
|
51 |
--policy.device=cuda \
|
52 |
--wandb.enable=true
|
53 |
-
```
|
|
|
30 |
```bash
|
31 |
python lerobot/scripts/train.py \
|
32 |
--policy.path=lerobot/smolvla_base \
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
33 |
--dataset.repo_id=TekbotRobotics/svla_so101_pickplace_flags_sorting \
|
34 |
+
--batch_size=8 \
|
35 |
+
--steps=2000 \
|
36 |
--output_dir=outputs/train/my_smolvla \
|
37 |
--job_name=my_smolvla_training \
|
38 |
--policy.device=cuda \
|
39 |
--wandb.enable=true
|
40 |
+
```
|