From 3cbd0ba7cc5555ed09d11183a1d0b1845973a0df Mon Sep 17 00:00:00 2001 From: shahules786 Date: Mon, 7 Nov 2022 13:00:34 +0530 Subject: [PATCH] mv coeff to device --- enhancer/utils/transforms.py | 1 + 1 file changed, 1 insertion(+) diff --git a/enhancer/utils/transforms.py b/enhancer/utils/transforms.py index fbdb8f9..2acbb08 100644 --- a/enhancer/utils/transforms.py +++ b/enhancer/utils/transforms.py @@ -85,6 +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 = F.conv_transpose1d(coeff, self.enframe, stride=self.hop_size) out = out / (coeff + 1e-8) pad = self.window_len - self.hop_size