From b8adeef2646d0890e59fa4b1a5247276ff2cb459 Mon Sep 17 00:00:00 2001
From: Eren Golge <egolge@mozilla.com>
Date: Fri, 9 Mar 2018 04:06:36 -0800
Subject: [PATCH] config change

---
 config.json | 4 ++--
 train.py    | 1 +
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/config.json b/config.json
index 838f151..3c43152 100644
--- a/config.json
+++ b/config.json
@@ -12,9 +12,9 @@
   "text_cleaner": "english_cleaners",
 
   "epochs": 2000,
-  "lr": 0.0006,
+  "lr": 0.0003,
   "warmup_steps": 4000,
-  "batch_size": 32,
+  "batch_size": 8,
   "r": 5,
 
   "griffin_lim_iters": 60,
diff --git a/train.py b/train.py
index 8aa6567..1b2c944 100644
--- a/train.py
+++ b/train.py
@@ -45,6 +45,7 @@ OUT_PATH = create_experiment_folder(OUT_PATH)
 CHECKPOINT_PATH = os.path.join(OUT_PATH, 'checkpoints')
 shutil.copyfile(args.config_path, os.path.join(OUT_PATH, 'config.json'))
 
+parser.add_argument('--finetine_path', type=str)
 # save config to tmp place to be loaded by subsequent modules.
 file_name = str(os.getpid())
 tmp_path = os.path.join("/tmp/", file_name+'_tts')