From 60fc4607d03c907d8832dc35fcbed363426da5d9 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Mon, 7 Nov 2022 10:24:18 +0530 Subject: [PATCH] init projection_size as None --- enhancer/models/complexnn/rnn.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/enhancer/models/complexnn/rnn.py b/enhancer/models/complexnn/rnn.py index 7d19425..847030b 100644 --- a/enhancer/models/complexnn/rnn.py +++ b/enhancer/models/complexnn/rnn.py @@ -42,6 +42,8 @@ class ComplexLSTM(nn.Module): self.imag_linear = nn.Linear( self.hidden_size * bidirectional, self.projection_size ) + else: + self.projection_size = None def forward(self, input):