From 921d9513c3719c7af98b11766ea9f985b8052ee5 Mon Sep 17 00:00:00 2001
From: Aditya NG <adityang5@gmail.com>
Date: Wed, 8 May 2024 10:48:31 +0530
Subject: [PATCH] feat(sweep): reduced batch size

---
 kan_gpt/sweep.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/kan_gpt/sweep.py b/kan_gpt/sweep.py
index d6b4f82..7bdf1be 100644
--- 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",
         ]