Update helper.py

helper.py

@@ -77,10 +77,10 @@ def evaluate(estimate, reference):
     print(f"STOI score: {stoi_mix}")
 
 def get_irms(stft_clean, stft_noise):
-    if stft_clean.size(1) > stft_noise.size(1):
-       stft_noise = F.pad(stft_noise, (0, stft_clean.size(1) - stft_noise.size(1)))
-    elif stft_noise.size(1) > stft_clean.size(1):
-        stft_clean = F.pad(stft_clean, (0, stft_noise.size(1) - stft_clean.size(1)))
+    if stft_clean.size(2) > stft_noise.size(2):
+       stft_noise = F.pad(stft_noise, (0, stft_clean.size(2) - stft_noise.size(2)))
+    elif stft_noise.size(2) > stft_clean.size(2):
+        stft_clean = F.pad(stft_clean, (0, stft_noise.size(2) - stft_clean.size(2)))
     mag_clean = stft_clean.abs() ** 2
     mag_noise = stft_noise.abs() ** 2
     irm_speech = mag_clean / (mag_clean + mag_noise)