Mirrors/TTS
1
0
Fork 0
mirror of https://github.com/coqui-ai/TTS.git synced 2025-12-25 12:49:29 +01:00
Code Issues Packages Projects Releases Wiki Activity

bug fix

Browse Source
This commit is contained in:
Eren Golge
2018-05-02 06:21:29 -07:00
parent 9924ae4a1d
commit 1fa6068e9d
2 changed files with 909 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

942
notebooks/TacotronPlayGround.ipynb
View File

File diff suppressed because one or more lines are too long

4
train.py
View File

@@ -94,7 +94,7 @@ def train(model, criterion, criterion_st, data_loader, optimizer, epoch):
optimizer.zero_grad()
stop_target = stop_target.view(c.batch_size, stop_target.size(1) // c.r, -1)
stop_target = stop_target.view(text_input.shape[0], stop_target.size(1) // c.r, -1)
stop_target = (stop_target.sum(2) > 0.0).float().unsqueeze(2)
# dispatch data to GPU
@@ -239,7 +239,7 @@ def evaluate(model, criterion, criterion_st, data_loader, current_step):
mel_lengths = data[4]
stop_target = data[5]
stop_target = stop_target.view(c.batch_size, stop_target.size(1) // c.r, -1)
stop_target = stop_target.view(text_input.shape[0], stop_target.size(1) // c.r, -1)
stop_target = (stop_target.sum(2) > 0.0).float().unsqueeze(2)
# dispatch data to GPU