update

examples/spectrum_unet_irm_aishell/step_2_train_model.py

@@ -298,6 +298,7 @@ def main():
             snr_loss = snr_mse_loss.forward(lsnr_prediction, snr_db_target)
             if torch.any(torch.isnan(snr_loss)):
                 raise AssertionError("nan in snr_loss")
+            print(f"irm_loss: {irm_loss}, snr_loss: {snr_loss}")
             loss = irm_loss + 0 * snr_loss
             # loss = irm_loss