From 2186429fa8fce1ec5ba1b60a9334bc33eb259d42 Mon Sep 17 00:00:00 2001
From: Cadene <re.cadene@gmail.com>
Date: Wed, 10 Apr 2024 15:00:54 +0000
Subject: [PATCH] policy.batch_size=2

---
 .github/workflows/test.yml | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 91319852..8d0b6851 100644
--- 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