From 40e8722014b0e12ef47638046975a00951000b70 Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Mon, 7 Nov 2022 10:52:35 +0530
Subject: [PATCH] fix o/p shape

---
 enhancer/models/dccrn.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/enhancer/models/dccrn.py b/enhancer/models/dccrn.py
index c6ee837..72d8a23 100644
--- a/enhancer/models/dccrn.py
+++ b/enhancer/models/dccrn.py
@@ -325,6 +325,6 @@ class DCCRN(Model):
             imag = imag * mask_imag
 
         spec = torch.cat([real, imag], 1)
-        wav = self.istft(spec).squeeze(1)
+        wav = self.istft(spec)
         wav = wav.clamp_(-1, 1)
         return wav