Update train.py

Reduced train epoch from 14 to 4 to check git action works for shorter duration, earlier it timed out for 14 epoch even container size of 956mb
The-School-of-AI · Sep 15, 2024 · ccaced6 · ccaced6
1 parent eb71a88
commit ccaced6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -74,7 +74,7 @@ def main():
                         help='input batch size for training (default: 64)')
     parser.add_argument('--test-batch-size', type=int, default=1000, metavar='N',
                         help='input batch size for testing (default: 1000)')
-    parser.add_argument('--epochs', type=int, default=14, metavar='N',
+    parser.add_argument('--epochs', type=int, default=4, metavar='N',
                         help='number of epochs to train (default: 14)')
     parser.add_argument('--lr', type=float, default=1.0, metavar='LR',
                         help='learning rate (default: 1.0)')