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():