Update losses.py

2023-12-20 02:50:58 +08:00
parent 97f18c0e26
commit 2e7c173de2
2 changed files with 12 additions and 9 deletions
--- a/losses.py
+++ b/losses.py
@@ -67,6 +67,8 @@ class WavLMLoss(torch.nn.Module):
        self.wd = wd
        self.resample = torchaudio.transforms.Resample(model_sr, slm_sr)
        self.wavlm.eval()
+        for param in self.wavlm.parameters():
+            param.requires_grad = False

    def forward(self, wav, y_rec):
        with torch.no_grad():