mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-23 12:40:08 +00:00
also shorten action_steps
This commit is contained in:
@@ -90,6 +90,7 @@ def test_pi05_forward_pass():
|
|||||||
dtype="float32",
|
dtype="float32",
|
||||||
pi05=True,
|
pi05=True,
|
||||||
action_horizon=16, # Shorter horizon for testing
|
action_horizon=16, # Shorter horizon for testing
|
||||||
|
n_action_steps=16, # Shorter action steps for testing
|
||||||
)
|
)
|
||||||
|
|
||||||
# Create dummy dataset stats
|
# Create dummy dataset stats
|
||||||
|
|||||||
Reference in New Issue
Block a user