return tensor
This commit is contained in:
parent
551ee4436f
commit
89ab58c5df
|
|
@ -126,7 +126,7 @@ class Pesq:
|
||||||
return self.pesq(prediction, target)
|
return self.pesq(prediction, target)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logging.warning(f"{e} error occured while calculating PESQ")
|
logging.warning(f"{e} error occured while calculating PESQ")
|
||||||
return 0.0
|
return torch.tensor(0.0)
|
||||||
|
|
||||||
|
|
||||||
class LossWrapper(nn.Module):
|
class LossWrapper(nn.Module):
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue