mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-20 02:59:50 +00:00
Add workable flow
This commit is contained in:
@@ -39,6 +39,12 @@ Usage:
|
|||||||
--rtc.execution_horizon=8 \
|
--rtc.execution_horizon=8 \
|
||||||
--device=cuda
|
--device=cuda
|
||||||
|
|
||||||
|
uv run python examples/rtc/eval_dataset.py \
|
||||||
|
--policy.path=lipsop/reuben_pi0 \
|
||||||
|
--dataset.repo_id=ReubenLim/so101_cube_in_cup \
|
||||||
|
--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
|
||||||
uv run python examples/rtc/eval_dataset.py \
|
uv run python examples/rtc/eval_dataset.py \
|
||||||
|
|||||||
Reference in New Issue
Block a user