chore(docs): update instructions for change in device and use_amp (#843)

This commit is contained in:
Steven Palma
2025-03-10 21:03:33 +01:00
committed by GitHub
parent 8a64d8268b
commit 12f52632ed
7 changed files with 43 additions and 43 deletions

View File

@@ -232,8 +232,8 @@ python lerobot/scripts/eval.py \
--env.type=pusht \
--eval.batch_size=10 \
--eval.n_episodes=10 \
--use_amp=false \
--device=cuda
--policy.use_amp=false \
--policy.device=cuda
```
Note: After training your own policy, you can re-evaluate the checkpoints with: