HoneyTian commited on
Commit
20fa6bf
·
1 Parent(s): cedfdcf
examples/conv_tasnet/step_2_train_model.py CHANGED
@@ -283,6 +283,7 @@ def main():
283
  neg_stoi_loss = neg_stoi_loss_fn.forward(denoise_audios, clean_audios)
284
  mr_stft_loss = mr_stft_loss_fn.forward(denoise_audios, clean_audios)
285
  pesq_loss = pesq_loss_fn.forward(clean_audios, denoise_audios)
 
286
 
287
  # loss = 0.25 * ae_loss + 0.25 * neg_si_snr_loss
288
  # loss = 0.25 * ae_loss + 0.25 * neg_si_snr_loss + 0.25 * neg_stoi_loss + 0.25 * mr_stft_loss
 
283
  neg_stoi_loss = neg_stoi_loss_fn.forward(denoise_audios, clean_audios)
284
  mr_stft_loss = mr_stft_loss_fn.forward(denoise_audios, clean_audios)
285
  pesq_loss = pesq_loss_fn.forward(clean_audios, denoise_audios)
286
+ print(f"pesq_loss: {pesq_loss}")
287
 
288
  # loss = 0.25 * ae_loss + 0.25 * neg_si_snr_loss
289
  # loss = 0.25 * ae_loss + 0.25 * neg_si_snr_loss + 0.25 * neg_stoi_loss + 0.25 * mr_stft_loss