Robotics
LeRobot
Safetensors
Kazeo57 commited on
Commit
4bbe671
·
verified ·
1 Parent(s): f451f66

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -16
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
- --batch_size=64 \
48
- --steps=200000 \
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
+ ```