Update README.md
Browse files
README.md
CHANGED
|
@@ -1,8 +1,4 @@
|
|
| 1 |
-
python -m lerobot.scripts.train --dataset.repo_id=DRMNmadhan/piper_pickdrop_200 --policy.path=lerobot/smolvla_base --batch_size=10 --steps=40000 --eval_freq=5000 --save_freq=5000 --policy.device=cuda --policy.push_to_hub=false --dataset.video_backend=pyav --wandb.enable=true --policy.n_obs_steps=2 --policy.optimizer_lr=2.5e-5 --policy.freeze_vision_encoder=false --policy.train_expert_only=false --policy.scheduler_decay_steps=40000 --wandb.enable=true
|
| 2 |
-
---
|
| 3 |
-
datasets:
|
| 4 |
-
- DRMNmadhan/piper_pickdrop_200
|
| 5 |
base_model:
|
| 6 |
- lerobot/smolvla_base
|
| 7 |
-
pipeline_tag: robotics
|
| 8 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
base_model:
|
| 2 |
- lerobot/smolvla_base
|
|
|
|
| 3 |
---
|
| 4 |
+
python -m lerobot.scripts.train --dataset.repo_id=DRMNmadhan/piper_pickdrop_200 --policy.path=lerobot/smolvla_base --batch_size=10 --steps=40000 --eval_freq=5000 --save_freq=5000 --policy.device=cuda --policy.push_to_hub=false --dataset.video_backend=pyav --wandb.enable=true --policy.n_obs_steps=2 --policy.optimizer_lr=2.5e-5 --policy.freeze_vision_encoder=false --policy.train_expert_only=false --policy.scheduler_decay_steps=40000 --wandb.enable=true
|