config change

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,

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')