From 8046918cc185e78ec2016e1bbd7e0f4468213ef0 Mon Sep 17 00:00:00 2001
From: Michael Waskom <michael@modal.com>
Date: Tue, 6 Feb 2024 13:50:20 -0500
Subject: [PATCH] Smaller batch size

---
 ci/prep_for_ci.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ci/prep_for_ci.py b/ci/prep_for_ci.py
index 6884a79..85654a3 100644
--- a/ci/prep_for_ci.py
+++ b/ci/prep_for_ci.py
@@ -11,8 +11,8 @@ def main(config: str, data: str):
         cfg = yaml.safe_load(f.read())
     cfg["sequence_len"] = 1024
     cfg["val_set_size"] = 100
-    cfg["eval_batch_size"] = 10
-    cfg["micro_batch_size"] = 10
+    cfg["eval_batch_size"] = 2
+    cfg["micro_batch_size"] = 2
     cfg["num_epochs"] = 2
     cfg.pop("eval_steps", None)
     with open(config, "w") as f: