Merge pull request #34 from modal-labs/richard/try-fix-ci

modal-labs · Feb 22, 2024 · 6a18b37 · 6a18b37
2 parents 73feaf8 + 897c461
commit 6a18b37
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/ci/prep_for_ci.py b/ci/prep_for_ci.py
@@ -12,12 +12,14 @@ def main(config: str, data: str):
 
     if cfg["sample_packing"]:
         train_set_size = 2048
+        num_epochs = 10
     else:
         train_set_size = 1024
+        num_epochs = 1
     val_set_size = 64
 
     cfg["val_set_size"] = val_set_size
-    cfg["num_epochs"] = 1
+    cfg["num_epochs"] = num_epochs
     cfg.pop("eval_steps", None)  # Evaluate once at the end of the epoch
     with open(config, "w") as f:
         yaml.dump(cfg, f)