Fix end-to-end ACT train on Aloha

2024-03-27 12:43:30 +01:00 · 2024-03-27 12:43:30 +01:00 · 404b8f8a75
parent 17c2bbbeb8
commit 404b8f8a75
1 changed files with 2 additions and 0 deletions
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@ -140,6 +140,8 @@ jobs:
            device=cpu \
            save_model=true \
            save_freq=2 \
+            horizon=20 \
+            policy.batch_size=2 \
            hydra.run.dir=tests/outputs/act/

      - name: Test eval ACT on ALOHA end-to-end