diff --git a/enhancer/loss.py b/enhancer/loss.py index 9ef90d2..fd1a609 100644 --- a/enhancer/loss.py +++ b/enhancer/loss.py @@ -126,7 +126,7 @@ class Pesq: return self.pesq(prediction, target) except Exception as e: logging.warning(f"{e} error occured while calculating PESQ") - return 0.0 + return torch.tensor(0.0) class LossWrapper(nn.Module):