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

Merge pull request #549 from SanjaESC/patch-1 

Update losses.py
This commit is contained in:
Eren Gölge 2020-11-02 12:43:24 +01:00 committed by GitHub
parent 587576d5ad 47d74ced1c
commit 0cd222d4b7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -169,7 +169,7 @@ class GuidedAttentionLoss(torch.nn.Module):
@staticmethod @staticmethod
def _make_ga_mask(ilen, olen, sigma): def _make_ga_mask(ilen, olen, sigma):
grid_x, grid_y = torch.meshgrid(torch.arange(olen), torch.arange(ilen)) grid_x, grid_y = torch.meshgrid(torch.arange(olen, device=olen.device), torch.arange(ilen, device=ilen.device))
grid_x, grid_y = grid_x.float(), grid_y.float() grid_x, grid_y = grid_x.float(), grid_y.float()
return 1.0 - torch.exp(-(grid_y / ilen - grid_x / olen)**2 / return 1.0 - torch.exp(-(grid_y / ilen - grid_x / olen)**2 /
(2 * (sigma**2))) (2 * (sigma**2)))
@ -306,4 +306,4 @@ class GlowTTSLoss(torch.nn.Module):
return_dict['loss'] = log_mle + loss_dur return_dict['loss'] = log_mle + loss_dur
return_dict['log_mle'] = log_mle return_dict['log_mle'] = log_mle
return_dict['loss_dur'] = loss_dur return_dict['loss_dur'] = loss_dur
return return_dict return return_dict