feat(sweep): reduced batch size

AdityaNG · May 8, 2024 · 921d951 · 921d951
1 parent 3a30ff4
commit 921d951
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kan_gpt/sweep.py b/kan_gpt/sweep.py
@@ -65,7 +65,7 @@ class SweepArgs:
         num_workers = [
             0,
         ]
-        batch_size = [1, 2, 4, 8, 12, 16]
+        batch_size = [1, 2, 4, 8]
         dataset = [
             "tinyshakespeare",
         ]