Spaces:
Running
Running
update
Browse files
toolbox/torchaudio/models/clean_unet/modeling_clean_unet.py
CHANGED
@@ -95,17 +95,17 @@ class DecoderBlock(nn.Module):
|
|
95 |
def forward(self, inputs: torch.Tensor):
|
96 |
# inputs shape: [batch_size, channel, num_samples]
|
97 |
x = self.conv(inputs)
|
98 |
-
print(x._version
|
99 |
|
100 |
x = self.glu(x)
|
101 |
-
print(x._version
|
102 |
|
103 |
x = self.convt(x)
|
104 |
-
print(x._version
|
105 |
if self.do_relu:
|
106 |
x = F.relu(x)
|
107 |
# x = self.relu(x)
|
108 |
-
print(x._version
|
109 |
|
110 |
return x
|
111 |
|
|
|
95 |
def forward(self, inputs: torch.Tensor):
|
96 |
# inputs shape: [batch_size, channel, num_samples]
|
97 |
x = self.conv(inputs)
|
98 |
+
print(x._version)
|
99 |
|
100 |
x = self.glu(x)
|
101 |
+
print(x._version)
|
102 |
|
103 |
x = self.convt(x)
|
104 |
+
print(x._version)
|
105 |
if self.do_relu:
|
106 |
x = F.relu(x)
|
107 |
# x = self.relu(x)
|
108 |
+
print(x._version)
|
109 |
|
110 |
return x
|
111 |
|