Refactor eval.py (#127)

2024-05-03 17:33:16 +01:00
parent b7b69fcc3d
commit bccee745c3
12 changed files with 457 additions and 298 deletions
--- a/lerobot/configs/policy/act.yaml
+++ b/lerobot/configs/policy/act.yaml
@@ -28,7 +28,8 @@ training:
    action: "[i / ${fps} for i in range(${policy.chunk_size})]"

 eval:
-  n_episodes:: 50
+  n_episodes: 50
+  batch_size: 50

 # See `configuration_act.py` for more details.
 policy: