From e4f13946e83f01172e0a82b2bf52ebc83bbfceec Mon Sep 17 00:00:00 2001 From: shahules786 Date: Wed, 19 Oct 2022 12:38:29 +0530 Subject: [PATCH] fix demucs output --- enhancer/models/demucs.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/enhancer/models/demucs.py b/enhancer/models/demucs.py index 86afb6c..e5fa945 100644 --- a/enhancer/models/demucs.py +++ b/enhancer/models/demucs.py @@ -212,6 +212,8 @@ class Demucs(Model): waveform = waveform.mean(dim=1, keepdim=True) std = waveform.std(dim=-1, keepdim=True) waveform = waveform / (self.floor + std) + else: + std = 1 length = waveform.shape[-1] x = F.pad(waveform, (0, self.get_padding_length(length) - length)) if self.hparams.resample > 1: @@ -244,7 +246,7 @@ class Demucs(Model): ) out = x[..., :length] - return out + return std * out def get_padding_length(self, input_length):