From f29f1c3945649f2ab3d9bc53d354f12a997f900c Mon Sep 17 00:00:00 2001
From: Joseph Omar <josephomar@protonmail.com>
Date: Sun, 4 Aug 2024 11:43:17 +0000
Subject: [PATCH] help
---
recipes/Voicebank/enhance/MetricGAN-U/train.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/recipes/Voicebank/enhance/MetricGAN-U/train.py b/recipes/Voicebank/enhance/MetricGAN-U/train.py
index a0c68acd4..0bbd7f837 100644
--- a/recipes/Voicebank/enhance/MetricGAN-U/train.py
+++ b/recipes/Voicebank/enhance/MetricGAN-U/train.py
@@ -109,6 +109,8 @@ def dnsmos_eval(predict, target):
pred_wav = pred_wav / max(abs(pred_wav))
pred_wav = pred_wav[np.newaxis, :]
+ print("predwav_type", type(pred_wav), "shape", pred_wav.shape)
+
scores = dnsmos_func.run(pred_wav, sr=16000)
score = scores["p808_mos"]
score = float(sigmoid(score)) # normalize the score to 0~1
@@ -145,6 +147,8 @@ def dnsmos_eval_valid(predict, target):
pred_wav = pred_wav.numpy()
pred_wav = pred_wav / max(abs(pred_wav))
+ print("predwav_type", type(pred_wav), "shape", pred_wav.shape)
+
scores = dnsmos_func.run(pred_wav, sr=16000)
score = scores["p808_mos"]
return score
--
GitLab