update

examples/spectrum_unet_irm_aishell/step_2_train_model.py

@@ -292,7 +292,7 @@ def main():
             speech_irm_prediction, lsnr_prediction = model.forward(mix_spec)
             irm_loss = irm_mse_loss.forward(speech_irm_prediction, speech_irm_target)
             snr_loss = snr_mse_loss.forward(lsnr_prediction, snr_db_target)
-            loss = irm_loss + 0.1 * snr_loss
+            loss = irm_loss + 0.01 * snr_loss
             # loss = irm_loss
 
             total_loss += loss.item()
@@ -327,7 +327,7 @@ def main():
                 speech_irm_prediction, lsnr_prediction = model.forward(mix_spec)
                 irm_loss = irm_mse_loss.forward(speech_irm_prediction, speech_irm_target)
                 snr_loss = snr_mse_loss.forward(lsnr_prediction, snr_db_target)
-                loss = irm_loss + 0.1 * snr_loss
+                loss = irm_loss + 0.01 * snr_loss
                 # loss = irm_loss
 
             total_loss += loss.item()