Test notebook update

2018-04-21 04:23:54 -07:00 · 2018-04-21 04:23:54 -07:00 · 154ec7ba24
parent 1b49adee63
commit 154ec7ba24
2 changed files with 79 additions and 554 deletions
--- a/layers/tacotron.py
+++ b/layers/tacotron.py
@ -325,4 +325,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[:, int(alignment.shape[1]/2):].sum() > 0.7
+            and alignment.data[:, alignment.shape[1] - int(alignment.shape[1]/9):].sum() > 0.7
--- a/notebooks/TacotronPlayGround.ipynb
+++ b/notebooks/TacotronPlayGround.ipynb