eel4759_classification

commit 88375c7767dbea45e9a1158f0aa180043a59ffae
parent 3d5c9e1658ea8c8b06d9acaeff847647b58fa63a
Author: Vineet Kumar <git@vineetk.net>
Date:   Tue, 21 Apr 2026 22:54:46 -0400

main: increase batch size

Diffstat:
M
main.py
 | 
2
+-

1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/main.py b/main.py
@@ -37,7 +37,7 @@ def parse_args():
     )
     parser.add_argument('--epochs', type=int, default=50, help='Max training epochs')
     parser.add_argument('--patience', type=int, default=10, help='Early stopping patience')
-    parser.add_argument('--batch-size', type=int, default=32, help='Training batch size')
+    parser.add_argument('--batch-size', type=int, default=256, help='Training batch size')
     parser.add_argument('--lr', type=float, default=1e-4, help='Learning rate')
     parser.add_argument('--weight-decay', type=float, default=1e-4, help='AdamW weight decay')
     parser.add_argument('--seed', type=int, default=42, help='Random seed')