decreased batch size for tests

luxonis · May 30, 2024 · 549c5b9 · 549c5b9
1 parent b5cb7a7
commit 549c5b9
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/integration/test_sanity.py b/tests/integration/test_sanity.py
@@ -22,6 +22,8 @@ def test_sanity(config_file):
         "1",
         "trainer.callbacks",
         "[]",
+        "trainer.batch_size",
+        "1",
     ]
     result = subprocess.run(
         ["luxonis_train", "train", "--config", f"configs/{config_file}", *opts],
@@ -80,6 +82,8 @@ def test_tuner():
             "[]",
             "tuner.n_trials",
             "4",
+            "trainer.batch_size",
+            "1",
         ],
     )
     assert result.returncode == 0