From 9fc0a3840acf829f9c59b58d6a8aead86afd352e Mon Sep 17 00:00:00 2001
From: Subodh Malgonde <subodh@Subodhs-MacBook-Air.local>
Date: Thu, 21 Dec 2017 10:37:37 +0530
Subject: [PATCH] Changed default epochs, learning rate and batch size

---
 main.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/main.py b/main.py
index b24cd8b..6c155f4 100644
--- a/main.py
+++ b/main.py
@@ -293,7 +293,7 @@ def run():
         '--num_epochs',
         type=int,
         nargs='?',
-        default=80,
+        default=20,
         help='Number of epochs.'
     )
     parser.add_argument(
@@ -301,7 +301,7 @@ def run():
         '--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.'
     )