"tests/nn/git@developer.sourcefind.cn:OpenDAS/fairscale.git" did not exist on "d8fc94d95d51e496c5f56543165b5a1fe4f81c2c"
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):
elif use_conv:
self.conv = nn.Conv1d(self.channels, self.out_channels, 3, padding=1)
def forward(self, x):
assert x.shape[1] == self.channels
def forward(self, inputs):
assert inputs.shape[1] == self.channels
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:
x = self.conv(x)
outputs = self.conv(outputs)
return x
return outputs
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