From 88375c7767dbea45e9a1158f0aa180043a59ffae Mon Sep 17 00:00:00 2001 From: Vineet Kumar Date: Tue, 21 Apr 2026 22:54:46 -0400 Subject: main: increase batch size --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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') -- cgit v1.2.3