HoneyTian commited on
Commit
ec8bf87
·
1 Parent(s): cdaf8e7
examples/clean_unet_aishell/step_2_train_model.py CHANGED
@@ -238,6 +238,7 @@ def main():
238
  enhanced_audios_list_r = list(enhanced_audios.detach().cpu().numpy())
239
  clean_audios_list_r = list(clean_audios.detach().cpu().numpy())
240
  pesq_metric = batch_pesq(enhanced_audios_list_r, clean_audios_list_r)
 
241
 
242
  optimizer.zero_grad()
243
  loss.backward()
 
238
  enhanced_audios_list_r = list(enhanced_audios.detach().cpu().numpy())
239
  clean_audios_list_r = list(clean_audios.detach().cpu().numpy())
240
  pesq_metric = batch_pesq(enhanced_audios_list_r, clean_audios_list_r)
241
+ print(f"pesq_metric: {pesq_metric}")
242
 
243
  optimizer.zero_grad()
244
  loss.backward()