From 97cce102503311fa82aef6d753623ad6108cee40 Mon Sep 17 00:00:00 2001
From: Eren G??lge <egolge@coqui.ai>
Date: Mon, 15 Aug 2022 10:42:13 +0200
Subject: [PATCH] Fix BCE loss issue

---
 TTS/tts/layers/losses.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/TTS/tts/layers/losses.py b/TTS/tts/layers/losses.py
index b9a03af1..dcb38d4a 100644
--- a/TTS/tts/layers/losses.py
+++ b/TTS/tts/layers/losses.py
@@ -1,3 +1,4 @@
+from importlib.metadata import requires
 import math
 
 import numpy as np
@@ -165,7 +166,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):
         """