init projection_size as None

2022-11-07 10:24:18 +05:30 · 2022-11-07 10:24:18 +05:30 · 60fc4607d0
parent c21f05e307
commit 60fc4607d0
1 changed files with 2 additions and 0 deletions
--- 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):