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 loss

This commit is contained in:
Eren Golge 2018-04-25 08:00:30 -07:00
parent 52b4bc6bed
commit e257bd7278
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
layers/losses.py
View File

@ -44,10 +44,11 @@ class L1LossMasked(nn.Module):
# target_flat: (batch * max_len, dim) # target_flat: (batch * max_len, dim)
target_flat = target.view(-1, target.shape[-1]) target_flat = target.view(-1, target.shape[-1])
# losses_flat: (batch * max_len, dim) # losses_flat: (batch * max_len, dim)
losses_flat = functional.l1_loss(input, target, size_average=False, losses_flat = functional.l1_loss(input, target_flat, size_average=False,
reduce=False) reduce=False)
# losses: (batch, max_len, dim) # losses: (batch, max_len, dim)
losses = losses_flat.view(*target.size()) losses = losses_flat.view(*target.size())
# mask: (batch, max_len, 1) # mask: (batch, max_len, 1)
mask = _sequence_mask(sequence_length=length, mask = _sequence_mask(sequence_length=length,
max_len=target.size(1)).unsqueeze(2) max_len=target.size(1)).unsqueeze(2)