update

toolbox/torchaudio/models/clean_unet/modeling_clean_unet.py

@@ -90,7 +90,7 @@ class DecoderBlock(nn.Module):
         self.conv = nn.Conv1d(channels_h, channels_h * 2, 1)
         self.glu = nn.GLU(dim=1)
         self.convt = nn.ConvTranspose1d(channels_h, channels_output, kernel_size, stride)
-        self.relu = nn.ReLU()
+        # self.relu = nn.ReLU()
 
     def forward(self, inputs: torch.Tensor):
         # inputs shape: [batch_size, channel, num_samples]
@@ -98,7 +98,8 @@ class DecoderBlock(nn.Module):
         x = self.glu(x)
         x = self.convt(x)
         if self.do_relu:
-            x = self.relu(x)
+            x = F.relu(x)
+            # x = self.relu(x)
         return x