mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-20 19:19:56 +00:00
Merge branch 'main' into feature/add-multitask-dit
This commit is contained in:
@@ -130,13 +130,13 @@ lerobot-train \
|
|||||||
--policy.path="lerobot/xvla-base" \
|
--policy.path="lerobot/xvla-base" \
|
||||||
--policy.repo_id="HF_USER/xvla-your-robot" \
|
--policy.repo_id="HF_USER/xvla-your-robot" \
|
||||||
--policy.dtype=bfloat16 \
|
--policy.dtype=bfloat16 \
|
||||||
--steps=3000 \
|
--policy.action_mode=auto \
|
||||||
|
--steps=20000 \
|
||||||
--policy.device=cuda \
|
--policy.device=cuda \
|
||||||
--policy.freeze_vision_encoder=false \
|
--policy.freeze_vision_encoder=false \
|
||||||
--policy.freeze_language_encoder=false \
|
--policy.freeze_language_encoder=false \
|
||||||
--policy.train_policy_transformer=true \
|
--policy.train_policy_transformer=true \
|
||||||
--policy.train_soft_prompts=true \
|
--policy.train_soft_prompts=true \
|
||||||
--policy.action_mode=YOUR_ACTION_MODE
|
|
||||||
```
|
```
|
||||||
|
|
||||||
### Training Parameters Explained
|
### Training Parameters Explained
|
||||||
|
|||||||
Reference in New Issue
Block a user