HoneyTian commited on
Commit
987be40
·
1 Parent(s): 852815e
toolbox/torchaudio/models/clean_unet/modeling_clean_unet.py CHANGED
@@ -144,7 +144,7 @@ class CleanUNet(nn.Module):
144
  nn.Conv1d(channels_h, channels_h * 2, 1),
145
  nn.GLU(dim=1),
146
  nn.ConvTranspose1d(channels_h, channels_output, kernel_size, stride),
147
- nn.ReLU(inplace=False)
148
  ))
149
  channels_output = channels_h
150
 
 
144
  nn.Conv1d(channels_h, channels_h * 2, 1),
145
  nn.GLU(dim=1),
146
  nn.ConvTranspose1d(channels_h, channels_output, kernel_size, stride),
147
+ # nn.ReLU(inplace=False)
148
  ))
149
  channels_output = channels_h
150