HoneyTian commited on
Commit
d085023
·
1 Parent(s): d5167a5
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()
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()
148
  ))
149
  channels_output = channels_h
150