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

fix o/p shape

This commit is contained in:
shahules786 2022-11-07 10:52:35 +05:30
parent 15c1d1ad94
commit 40e8722014
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/models/dccrn.py
View File

@ -325,6 +325,6 @@ class DCCRN(Model):
imag = imag * mask_imag imag = imag * mask_imag
spec = torch.cat([real, imag], 1) spec = torch.cat([real, imag], 1)
wav = self.istft(spec).squeeze(1) wav = self.istft(spec)
wav = wav.clamp_(-1, 1) wav = wav.clamp_(-1, 1)
return wav return wav