refactor

wenet-e2e · Mar 2, 2024 · 6af3a8d · 6af3a8d
1 parent f56726b
commit 6af3a8d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/wenet/dataset/processor.py b/wenet/dataset/processor.py
@@ -310,8 +310,8 @@ def compute_log_mel_spectrogram(sample,
         waveform = F.pad(waveform, (0, padding))
     if pad_or_trim:
         length = max_duration * sample_rate
-        if waveform.size(1) >= length:
-            waveform = waveform[:, :length]
+        if waveform.size() >= length:
+            waveform = waveform[:length]
         else:
             waveform = F.pad(waveform, (0, length - waveform.size(1)))