also shorten action_steps

2026-05-15 08:39:49 +00:00 · 2025-09-10 21:36:58 +02:00
parent ac323b0113
commit e9e7eb827a
1 changed files with 1 additions and 0 deletions
@@ -90,6 +90,7 @@ def test_pi05_forward_pass():
        dtype="float32",
        pi05=True,
        action_horizon=16,  # Shorter horizon for testing
+        n_action_steps=16,  # Shorter action steps for testing
    )

    # Create dummy dataset stats