wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-09 11:16:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

bug fix loss

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
layers/losses.py
+2 -1
View File
@@ -44,10 +44,11 @@ class L1LossMasked(nn.Module):
# target_flat: (batch * max_len, dim)
target_flat = target.view(-1, target.shape[-1])
# 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)
# losses: (batch, max_len, dim)
losses = losses_flat.view(*target.size())
# mask: (batch, max_len, 1)
mask = _sequence_mask(sequence_length=length,
max_len=target.size(1)).unsqueeze(2)