update

examples/spectrum_unet_irm_aishell/step_2_train_model.py

@@ -304,7 +304,7 @@ def main():
             if torch.any(torch.isnan(snr_loss)) or torch.any(torch.isinf(snr_loss)):
                 raise AssertionError("nan or inf in snr_loss")
             # loss = irm_loss + 0.1 * snr_loss
-            loss = irm_loss + 0.05 * snr_loss
+            loss = 10.0 * irm_loss + 0.05 * snr_loss
             # loss = irm_loss
 
             total_loss += loss.item()
@@ -347,7 +347,7 @@ def main():
                     raise AssertionError(f"expected lsnr_prediction between 0 and 1.")
                 snr_loss = snr_mse_loss.forward(lsnr_prediction, snr_db_target)
                 # loss = irm_loss + 0.1 * snr_loss
-                loss = irm_loss + 0.05 * snr_loss
+                loss = 10.0 * irm_loss + 0.05 * snr_loss
                 # loss = irm_loss
 
                 total_loss += loss.item()