fix typo in config parser methods

vijuSR · Feb 1, 2019 · 964a395 · 964a395
1 parent f9859d5
commit 964a395
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/trainer.py b/src/trainer.py
@@ -118,10 +118,10 @@ def test(emoji_data, sess):
     if not os.path.exists(CHECKPOINT_SAVE_PATH):
         os.makedirs(CHECKPOINT_SAVE_PATH)
 
-    BATCH_SIZE = config_parser.getInt('MODEL_HYPER_PARAMETERS', 'batch_size')
-    STEPS = config_parser.getInt('MODEL_HYPER_PARAMETERS', 'train_steps')
-    LEARNING_RATE = config_parser.getFloat('MODEL_HYPER_PARAMETERS', 'learning_rate')
-    KEEP_PROB = config_parser.getFloat('MODEL_HYPER_PARAMETERS', 'dropout_keep_prob')
+    BATCH_SIZE = config_parser.getint('MODEL_HYPER_PARAMETERS', 'batch_size')
+    STEPS = config_parser.getint('MODEL_HYPER_PARAMETERS', 'train_steps')
+    LEARNING_RATE = config_parser.getfloat('MODEL_HYPER_PARAMETERS', 'learning_rate')
+    KEEP_PROB = config_parser.getfloat('MODEL_HYPER_PARAMETERS', 'dropout_keep_prob')
 
     X = tf.placeholder(
         tf.float32, shape=[None, 48, 48, 1]