From ed210a8c60ef9f35c1dc243ab71873c80da7a17f Mon Sep 17 00:00:00 2001 From: shahules786 Date: Mon, 7 Nov 2022 16:00:47 +0530 Subject: [PATCH] mv coeff to input 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 2acbb08..f8e4b50 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(self.device) + 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