update

examples/dfnet/step_2_train_model.py

@@ -263,6 +263,7 @@ def main():
 
             est_spec, est_wav, est_mask, lsnr = model.forward(noisy_audios)
 
+            print(f"est_mask.shape: {est_mask.shape}")
             neg_si_snr_loss = neg_si_snr_loss_fn.forward(est_wav, clean_audios)
             mask_loss = model.mask_loss_fn(est_mask, clean_audios, noisy_audios)
             # mr_stft_loss = mr_stft_loss_fn.forward(denoise_audios, clean_audios)