Spaces:
Running
Running
update
Browse files
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 |
|