update

examples/clean_unet_aishell/step_2_train_model.py

@@ -238,14 +238,17 @@ def main():
             enhanced_audios_list_r = list(enhanced_audios.detach().cpu().numpy())
             clean_audios_list_r = list(clean_audios.detach().cpu().numpy())
             pesq_metric = batch_pesq(enhanced_audios_list_r, clean_audios_list_r)
-            print(f"pesq_metric: {pesq_metric}")
+            if pesq_metric is None:
+                pesq_metric = 0
+            else:
+                pesq_metric = torch.mean(pesq_metric).item()
 
             optimizer.zero_grad()
             loss.backward()
             optimizer.step()
             lr_scheduler.step()
 
-            total_pesq_metric += 0 if pesq_metric is None else pesq_metric.item()
+            total_pesq_metric += pesq_metric
             total_loss += loss.item()
             total_ae_loss += ae_loss.item()
             total_sc_loss += sc_loss.item()