update

examples/dtln/step_2_train_model.py

@@ -323,6 +323,7 @@ def main():
                         noisy_audios: torch.Tensor = noisy_audios.to(device)
 
                         denoise_audios = model.forward(noisy_audios)
+                        denoise_audios = torch.squeeze(denoise_audios, dim=1)
 
                         mr_stft_loss = mr_stft_loss_fn.forward(denoise_audios, clean_audios)
                         neg_si_snr_loss = neg_si_snr_loss_fn.forward(denoise_audios, clean_audios)