HoneyTian commited on
Commit
c3c3d31
·
1 Parent(s): 9de2c38
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(clean_audios, denoise_audios.detach())
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: