From 82ffe819eb4bba92682a68c29a1ee3f7ed0fe192 Mon Sep 17 00:00:00 2001
From: Eren Golge <egolge@mozilla.com>
Date: Sun, 29 Apr 2018 06:12:14 -0700
Subject: [PATCH] thweb finetune

---
 config.json        | 10 +++++-----
 layers/tacotron.py |  2 +-
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/config.json b/config.json
index e6c2af11..df3b3372 100644
--- a/config.json
+++ b/config.json
@@ -12,7 +12,7 @@
     "text_cleaner": "english_cleaners",
 
     "epochs": 50,
-    "lr": 0.002,
+    "lr": 0.004,
     "warmup_steps": 4000,
     "batch_size": 32,
     "eval_batch_size":32,
@@ -23,10 +23,10 @@
     "griffin_lim_iters": 60,
     "power": 1.2,
 
-    "dataset": "LJSpeech",
-    "meta_file_train": "metadata_train.csv",
-    "meta_file_val": "metadata_val.csv",
-    "data_path": "/data/shared/KeithIto/LJSpeech-1.0/",
+    "dataset": "TWEB",
+    "meta_file_train": "transcript_train.txt",
+    "meta_file_val": "transcript_val.txt",
+    "data_path": "/data/shared/BibleSpeech/",
     "min_seq_len": 0, 
     "num_loader_workers": 8,
 
diff --git a/layers/tacotron.py b/layers/tacotron.py
index aa8531ca..adf58b34 100644
--- a/layers/tacotron.py
+++ b/layers/tacotron.py
@@ -324,4 +324,4 @@ class Decoder(nn.Module):
 
 def is_end_of_frames(output, alignment, eps=0.01):  # 0.2
     return ((output.data <= eps).prod(0) > 0).any() \
-            and alignment.data[:, alignment.shape[1] - int(alignment.shape[1]/9):].sum() > 0.7
\ No newline at end of file
+            and alignment.data[:, -10:].sum() > 0.7
\ No newline at end of file