Fix: wavlm shape bug when batch_size 1

losses.py

@@ -2,6 +2,8 @@ import torch
 import torchaudio
 from transformers import AutoModel
 
+from common.log import logger
+
 
 def feature_loss(fmap_r, fmap_g):
     loss = 0
@@ -78,7 +80,7 @@ class WavLMLoss(torch.nn.Module):
             ).hidden_states
         y_rec_16 = self.resample(y_rec)
         y_rec_embeddings = self.wavlm(
-            input_values=y_rec_16.squeeze(), output_hidden_states=True
+            input_values=y_rec_16, output_hidden_states=True
         ).hidden_states
 
         floss = 0