smaller training sets for sample_packing

modal-labs · Feb 15, 2024 · aaa3b68 · aaa3b68
1 parent 0501c25
commit aaa3b68
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,9 +11,9 @@ def main(config: str, data: str):
         cfg = yaml.safe_load(f.read())
 
     if cfg["sample_packing"]:
-        train_set_size = 4000
+        train_set_size = 2048
     else:
-        train_set_size = 1000
+        train_set_size = 1024
     val_set_size = 64
 
     cfg["val_set_size"] = val_set_size