update

toolbox/torchaudio/models/clean_unet/modeling_clean_unet.py

@@ -95,11 +95,18 @@ class DecoderBlock(nn.Module):
     def forward(self, inputs: torch.Tensor):
         # inputs shape: [batch_size, channel, num_samples]
         x = self.conv(inputs)
+        print(x._version())
+
         x = self.glu(x)
+        print(x._version())
+
         x = self.convt(x)
+        print(x._version())
         if self.do_relu:
             x = F.relu(x)
             # x = self.relu(x)
+        print(x._version())
+
         return x