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

init projection_size as None

This commit is contained in:
shahules786 2022-11-07 10:24:18 +05:30
parent c21f05e307
commit 60fc4607d0
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
enhancer/models/complexnn/rnn.py
View File

@ -42,6 +42,8 @@ class ComplexLSTM(nn.Module):
self.imag_linear = nn.Linear( self.imag_linear = nn.Linear(
self.hidden_size * bidirectional, self.projection_size self.hidden_size * bidirectional, self.projection_size
) )
else:
self.projection_size = None
def forward(self, input): def forward(self, input):