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

reset the way ga_loss is stored in return_dict

This commit is contained in:
erogol 2020-11-02 13:18:56 +01:00
parent a108d0ee81
commit d94782a076
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
TTS/tts/layers/losses.py
View File

@ -346,7 +346,7 @@ class TacotronLoss(torch.nn.Module):
if self.config.ga_alpha > 0: if self.config.ga_alpha > 0:
ga_loss = self.criterion_ga(alignments, input_lens, alignment_lens) ga_loss = self.criterion_ga(alignments, input_lens, alignment_lens)
loss += ga_loss * self.ga_alpha loss += ga_loss * self.ga_alpha
return_dict['ga_loss'] = ga_loss * self.ga_alpha return_dict['ga_loss'] = ga_loss
# decoder differential spectral loss # decoder differential spectral loss
if self.config.decoder_diff_spec_alpha > 0: if self.config.decoder_diff_spec_alpha > 0: