Spaces:
Running
Running
update
Browse files
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
|