Spaces:
Running
Running
update
Browse files
examples/conv_tasnet_gan/step_2_train_model.py
CHANGED
@@ -342,7 +342,7 @@ def main():
|
|
342 |
pesq_score_list: List[float] = run_batch_pesq(clean_audio_list, enhanced_audio_list, sample_rate=config.sample_rate, mode="nb")
|
343 |
|
344 |
metric_r = discriminator.forward(clean_audios, clean_audios)
|
345 |
-
metric_g = discriminator.forward(
|
346 |
loss_disc_r = F.mse_loss(one_labels, metric_r.flatten())
|
347 |
|
348 |
if -1 in pesq_score_list:
|
|
|
342 |
pesq_score_list: List[float] = run_batch_pesq(clean_audio_list, enhanced_audio_list, sample_rate=config.sample_rate, mode="nb")
|
343 |
|
344 |
metric_r = discriminator.forward(clean_audios, clean_audios)
|
345 |
+
metric_g = discriminator.forward(denoise_audios.detach(), clean_audios)
|
346 |
loss_disc_r = F.mse_loss(one_labels, metric_r.flatten())
|
347 |
|
348 |
if -1 in pesq_score_list:
|