From e941235ec0cda9cdb37b197157e8bc9b91797184 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Thu, 10 Nov 2022 10:34:48 +0530 Subject: [PATCH] mv coeff to device --- enhancer/utils/transforms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/enhancer/utils/transforms.py b/enhancer/utils/transforms.py index f8e4b50..5af1f92 100644 --- a/enhancer/utils/transforms.py +++ b/enhancer/utils/transforms.py @@ -85,7 +85,7 @@ class ConviSTFT(ConvFFT): input = torch.cat([real, imag], 1) out = F.conv_transpose1d(input, self.weight, stride=self.hop_size) coeff = self.window.unsqueeze(1).repeat(1, 1, input.size(-1)) ** 2 - coeff.to(input.device) + coeff = coeff.to(input.device) coeff = F.conv_transpose1d(coeff, self.enframe, stride=self.hop_size) out = out / (coeff + 1e-8) pad = self.window_len - self.hop_size