From 9cd9dfac6e43c57c03d19e3a0f441d4df7430820 Mon Sep 17 00:00:00 2001
From: Sihan Wang <wangsihan1995@gmail.com>
Date: Wed, 20 Dec 2023 00:07:15 +0800
Subject: [PATCH] Update losses.py

---
 losses.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/losses.py b/losses.py
index 53357aa..6a4dafa 100644
--- 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():