admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-24 02:06:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix ACT policy type examples in docs (#3792)

Browse Source
This commit is contained in:
someone114514
2026-06-24 23:59:07 -07:00
committed by GitHub
parent 536b9621b2
commit 508d18f8a1
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+1 -1
View File
@@ -97,7 +97,7 @@ Training a policy is as simple as running a script configuration:
```bash ```bash
lerobot-train \ lerobot-train \
--policy=act \ --policy.type=act \
--dataset.repo_id=lerobot/aloha_mobile_cabinet --dataset.repo_id=lerobot/aloha_mobile_cabinet
``` ```
docs/source/multi_gpu_training.mdx
+2 -2
View File
@@ -95,7 +95,7 @@ If you want to scale your hyperparameters when using multiple GPUs, you should d
accelerate launch --num_processes=2 $(which lerobot-train) \ accelerate launch --num_processes=2 $(which lerobot-train) \
--optimizer.lr=2e-4 \ --optimizer.lr=2e-4 \
--dataset.repo_id=lerobot/pusht \ --dataset.repo_id=lerobot/pusht \
--policy=act --policy.type=act
``` ```
**Training Steps Scaling:** **Training Steps Scaling:**
@@ -110,7 +110,7 @@ accelerate launch --num_processes=2 $(which lerobot-train) \
--batch_size=8 \ --batch_size=8 \
--steps=50000 \ --steps=50000 \
--dataset.repo_id=lerobot/pusht \ --dataset.repo_id=lerobot/pusht \
--policy=act --policy.type=act
``` ```
## Training Large Models with FSDP ## Training Large Models with FSDP