update

examples/conv_tasnet/step_2_train_model.py

@@ -195,6 +195,7 @@ def main():
         step_idx = int(step_idx)
         if step_idx > last_step_idx:
             last_step_idx = step_idx
+    last_epoch = 1
 
     if last_step_idx != -1:
         logger.info(f"resume from steps-{last_step_idx}.")
@@ -291,7 +292,7 @@ def main():
             neg_si_snr_loss = neg_si_snr_loss_fn.forward(denoise_audios, clean_audios)
             neg_stoi_loss = neg_stoi_loss_fn.forward(denoise_audios, clean_audios)
             mr_stft_loss = mr_stft_loss_fn.forward(denoise_audios, clean_audios)
-            pesq_loss = pesq_loss_fn.forward(clean_audios, denoise_audios)
+            pesq_loss = pesq_loss_fn.forward(denoise_audios, clean_audios)
 
             # loss = 0.25 * ae_loss + 0.25 * neg_si_snr_loss
             # loss = 0.25 * ae_loss + 0.25 * neg_si_snr_loss + 0.25 * neg_stoi_loss + 0.25 * mr_stft_loss
@@ -372,7 +373,7 @@ def main():
                         neg_si_snr_loss = neg_si_snr_loss_fn.forward(denoise_audios, clean_audios)
                         neg_stoi_loss = neg_stoi_loss_fn.forward(denoise_audios, clean_audios)
                         mr_stft_loss = mr_stft_loss_fn.forward(denoise_audios, clean_audios)
-                        pesq_loss = pesq_loss_fn.forward(clean_audios, denoise_audios)
+                        pesq_loss = pesq_loss_fn.forward(denoise_audios, clean_audios)
 
                         # loss = 0.25 * ae_loss + 0.25 * neg_si_snr_loss
                         # loss = 0.25 * ae_loss + 0.25 * neg_si_snr_loss + 0.25 * neg_stoi_loss + 0.25 * mr_stft_loss