bin.xue yingda.chen 3 years ago
parent
commit
6818ffdc8e
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      modelscope/metrics/audio_noise_metric.py

+ 3
- 1
modelscope/metrics/audio_noise_metric.py View File

@@ -35,6 +35,8 @@ class AudioNoiseMetric(Metric):
total_loss = avg_loss + avg_amp + avg_phase + avg_sisnr total_loss = avg_loss + avg_amp + avg_phase + avg_sisnr
return { return {
'total_loss': total_loss.item(), '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() MetricKeys.AVERAGE_LOSS: avg_loss.item()
} }

Loading…
Cancel
Save