policy.batch_size=2

2024-04-10 15:00:54 +00:00
parent 693f620df0
commit 2186429fa8
1 changed files with 2 additions and 0 deletions
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -184,6 +184,7 @@ jobs:
            device=cpu \
            save_model=true \
            save_freq=2 \
+            policy.batch_size=2 \
            hydra.run.dir=tests/outputs/diffusion/

      - name: Test eval Diffusion on PushT end-to-end
@@ -220,6 +221,7 @@ jobs:
            device=cpu \
            save_model=true \
            save_freq=2 \
+            policy.batch_size=2 \
            hydra.run.dir=tests/outputs/tdmpc/

      - name: Test eval TDMPC on Simxarm end-to-end