update

examples/spectrum_dfnet_aishell/step_2_train_model.py

@@ -327,7 +327,8 @@ 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 = speech_loss + irm_loss + snr_loss
+            # loss = speech_loss + irm_loss + snr_loss
+            loss = irm_loss
 
             total_loss += loss.item()
             total_examples += mix_complex_spec.size(0)
@@ -371,7 +372,8 @@ def main():
                 irm_loss = irm_mse_loss.forward(speech_irm_prediction, speech_irm_target)
                 snr_loss = snr_mse_loss.forward(lsnr_prediction, snr_db_target)
 
-                loss = speech_loss + irm_loss + snr_loss
+                # loss = speech_loss + irm_loss + snr_loss
+                loss = irm_loss
 
                 total_loss += loss.item()
             total_examples += mix_complex_spec.size(0)