Fix BCE loss issue (coqui-ai#1872)

* Fix BCE loss issue * Remove import
akshhack · Aug 15, 2022 · 4333492 · 4333492
1 parent c30b648
commit 4333492
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/TTS/tts/layers/losses.py b/TTS/tts/layers/losses.py
@@ -165,7 +165,7 @@ class BCELossMasked(nn.Module):
 
     def __init__(self, pos_weight: float = None):
         super().__init__()
-        self.pos_weight = torch.tensor([pos_weight])
+        self.pos_weight = nn.Parameter(torch.tensor([pos_weight]), requires_grad=False)
 
     def forward(self, x, target, length):
         """