Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'dev' of https://github.com/shahules786/enhancer into dev-hawk

This commit is contained in:
shahules786 2022-11-07 16:01:10 +05:30
parent 846b64ab88 ed210a8c60
commit 24fa16ca25
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
enhancer/utils/transforms.py
View File

@ -85,7 +85,7 @@ class ConviSTFT(ConvFFT):
input = torch.cat([real, imag], 1) input = torch.cat([real, imag], 1)
out = F.conv_transpose1d(input, self.weight, stride=self.hop_size) out = F.conv_transpose1d(input, self.weight, stride=self.hop_size)
coeff = self.window.unsqueeze(1).repeat(1, 1, input.size(-1)) ** 2 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) coeff = F.conv_transpose1d(coeff, self.enframe, stride=self.hop_size)
out = out / (coeff + 1e-8) out = out / (coeff + 1e-8)
pad = self.window_len - self.hop_size pad = self.window_len - self.hop_size