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 for losses

Browse Source
This commit is contained in:
root
2020-02-19 18:27:02 +01:00
committed by erogol
parent 34d2e9438d
commit 678d56cdef
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
layers/losses.py
+2
View File
@@ -37,6 +37,7 @@ class L1LossMasked(nn.Module):
x * mask, target * mask, reduction='none')
loss = loss.mul(out_weights.cuda()).sum()
else:
mask = mask.expand_as(x)
loss = functional.l1_loss(
x * mask, target * mask, reduction='sum')
loss = loss / mask.sum()
@@ -75,6 +76,7 @@ class MSELossMasked(nn.Module):
x * mask, target * mask, reduction='none')
loss = loss.mul(out_weights.cuda()).sum()
else:
mask = mask.expand_as(x)
loss = functional.mse_loss(
x * mask, target * mask, reduction='sum')
loss = loss / mask.sum()