admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-20 19:19:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

Pi0 eval dataset

Browse Source
This commit is contained in:
Eugene Mironov
2025-11-07 03:03:32 +07:00
parent d10b7787eb
commit e86afc883e
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/rtc/eval_dataset.py
+9 -2
View File
@@ -25,12 +25,19 @@ Usage:
--rtc.execution_horizon=8 \ --rtc.execution_horizon=8 \
--device=mps --device=mps
# Basic usage with pi0.5 policy # Basic usage with pi0.5 policy with cuda device
uv run python examples/rtc/eval_dataset.py \ uv run python examples/rtc/eval_dataset.py \
--policy.path=lerobot/pi05_libero_finetuned \ --policy.path=lerobot/pi05_libero_finetuned \
--dataset.repo_id=HuggingFaceVLA/libero \ --dataset.repo_id=HuggingFaceVLA/libero \
--rtc.execution_horizon=8 \ --rtc.execution_horizon=8 \
--device=mps --device=cuda
# Basic usage with pi0 policy with cuda device
uv run python examples/rtc/eval_dataset.py \
--policy.path=lerobot/pi0_libero_finetuned \
--dataset.repo_id=HuggingFaceVLA/libero \
--rtc.execution_horizon=8 \
--device=cuda
# With torch.compile for faster inference (PyTorch 2.0+) # With torch.compile for faster inference (PyTorch 2.0+)
# Note: CUDA graphs disabled by default due to in-place ops in denoising loop # Note: CUDA graphs disabled by default due to in-place ops in denoising loop