Update losses.py

2023-12-20 00:07:15 +08:00
parent 50ad3389e8
commit 9cd9dfac6e
1 changed files with 1 additions and 0 deletions
--- a/losses.py
+++ b/losses.py
@@ -66,6 +66,7 @@ class WavLMLoss(torch.nn.Module):
        self.wavlm = AutoModel.from_pretrained(model)
        self.wd = wd
        self.resample = torchaudio.transforms.Resample(model_sr, slm_sr)
+        self.wavlm.eval()

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