Merge branch 'dev' of https://github.com/shahules786/enhancer into dev-hawk

This commit is contained in:
shahules786 2022-11-07 12:02:42 +05:30
commit 2d0b309b4d
1 changed files with 2 additions and 0 deletions

View File

@ -3,6 +3,7 @@ import logging
import numpy as np
import torch
import torch.nn as nn
from torchmetrics import ScaleInvariantSignalNoiseRatio
from torchmetrics.audio.pesq import PerceptualEvaluationSpeechQuality
from torchmetrics.audio.stoi import ShortTimeObjectiveIntelligibility
@ -188,4 +189,5 @@ LOSS_MAP = {
"si-sdr": Si_SDR,
"pesq": Pesq,
"stoi": Stoi,
"si-snr": ScaleInvariantSignalNoiseRatio,
}