Changed default epochs, learning rate and batch size

tcotte · Dec 21, 2017 · 9fc0a38 · 9fc0a38
1 parent 3c90f08
commit 9fc0a38
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/main.py b/main.py
@@ -293,15 +293,15 @@ def run():
         '--num_epochs',
         type=int,
         nargs='?',
-        default=80,
+        default=20,
         help='Number of epochs.'
     )
     parser.add_argument(
         '-lr',
         '--learning_rate',
         type=float,
         nargs='?',
-        default=0.06,
+        default=0.001,
         help='Learning rate'
     )
 
@@ -319,7 +319,7 @@ def run():
         '--batch_size',
         type=int,
         nargs='?',
-        default=16,
+        default=8,
         help='Batch size.'
     )