Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

bug fix, average mel spec validation loss

This commit is contained in:
Eren Golge 2018-05-15 07:13:46 -07:00
parent 2c9ad3850a
commit 7d5fe41ab4
3 changed files with 277 additions and 147 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
layers/tacotron.py
View File

@ -315,7 +315,7 @@ class Decoder(nn.Module):
if t >= T_decoder: if t >= T_decoder:
break break
else: else:
if t > 1 and stop_token > 0.5: if t > 1 and stop_token > 0.8:
break break
elif t > self.max_decoder_steps: elif t > self.max_decoder_steps:
print(" !! Decoder stopped with 'max_decoder_steps'. \ print(" !! Decoder stopped with 'max_decoder_steps'. \

421
notebooks/TacotronPlayGround.ipynb
View File

File diff suppressed because one or more lines are too long

1
train.py
View File

@ -300,6 +300,7 @@ def evaluate(model, criterion, criterion_st, data_loader, current_step):
# compute average losses # compute average losses
avg_linear_loss /= (num_iter + 1) avg_linear_loss /= (num_iter + 1)
avg_mel_loss /= (num_iter + 1)
avg_stop_loss /= (num_iter + 1) avg_stop_loss /= (num_iter + 1)
avg_total_loss = avg_mel_loss + avg_linear_loss + stop_loss avg_total_loss = avg_mel_loss + avg_linear_loss + stop_loss