main: increase batch size

1 files changed, 1 insertions, 1 deletions

diff --git a/main.py b/main.py
index 6c4cd20..0d29d63 100644
--- 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')