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

rename action_horizon to chunk_size

Browse Source
This commit is contained in:
Pepijn
2025-09-11 19:42:25 +02:00
parent b044f3104b
commit 2234b851c0
8 changed files with 20 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test_pi0_pi05_hub.py
+2 -2
View File
@@ -82,7 +82,7 @@ def test_hub_loading(model_id="pepijn223/pi0_base_fp32", model_name="PI0"):
print(f" - Action expert variant: {policy.config.action_expert_variant}")
print(f" - Action dimension: {policy.config.action_dim}")
print(f" - State dimension: {policy.config.state_dim}")
print(f" - Action horizon: {policy.config.action_horizon}")
print(f" - Chunk_size: {policy.config.chunk_size}")
print(f" - Tokenizer max length: {policy.config.tokenizer_max_length}")
if model_name == "PI0.5":
print(f" - discrete_state_input: {policy.config.discrete_state_input}")
@@ -172,7 +172,7 @@ def test_hub_loading(model_id="pepijn223/pi0_base_fp32", model_name="PI0"):
),
"action": torch.randn(
batch_size,
policy.config.action_horizon,
policy.config.chunk_size,
policy.config.action_dim,
dtype=torch.float32,
device=device,