[to #42322933] feat: optimize ANS metric value

Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10399100
This commit is contained in:
bin.xue
2022-10-13 19:42:19 +08:00
committed by Yingda Chen
parent f18d46f727
commit 2bc90e2be1

View File

@@ -35,6 +35,8 @@ class AudioNoiseMetric(Metric):
total_loss = avg_loss + avg_amp + avg_phase + avg_sisnr
return {
'total_loss': total_loss.item(),
'avg_sisnr': avg_sisnr.item(),
# model use opposite number of sisnr as a calculation shortcut.
# revert it in evaluation result
'avg_sisnr': -avg_sisnr.item(),
MetricKeys.AVERAGE_LOSS: avg_loss.item()
}