Smaller batch size

modal-labs · Feb 6, 2024 · 8046918 · 8046918
1 parent 89ab93e
commit 8046918
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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: