HoneyTian commited on
Commit
a1cf85f
·
1 Parent(s): 6658511
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