Unverified Commit ffa33d63 authored by vikasmech's avatar vikasmech Committed by GitHub
Browse files

renamed variable to input_ and output_ (#3507)

* renamed variable to input_ and output_

* changed input
_ to intputs and output_ to outputs
parent d8ce53a8
...@@ -52,17 +52,17 @@ class Upsample1D(nn.Module): ...@@ -52,17 +52,17 @@ class Upsample1D(nn.Module):
elif use_conv: elif use_conv:
self.conv = nn.Conv1d(self.channels, self.out_channels, 3, padding=1) self.conv = nn.Conv1d(self.channels, self.out_channels, 3, padding=1)
def forward(self, x): def forward(self, inputs):
assert x.shape[1] == self.channels assert inputs.shape[1] == self.channels
if self.use_conv_transpose: if self.use_conv_transpose:
return self.conv(x) return self.conv(inputs)
x = F.interpolate(x, scale_factor=2.0, mode="nearest") outputs = F.interpolate(inputs, scale_factor=2.0, mode="nearest")
if self.use_conv: if self.use_conv:
x = self.conv(x) outputs = self.conv(outputs)
return x return outputs
class Downsample1D(nn.Module): class Downsample1D(nn.Module):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment