Skip to content

GitLab

Unverified Commit 2937d77d authored by Francisco Massa's avatar Francisco Massa Committed by GitHub
Browse files

Fix torchscript issue in ConvTranspose2d (#1917)

parent 21603531
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/ops/misc.py
View file @ 2937d77d
...@@ -42,7 +42,7 @@ class ConvTranspose2d(torch.nn.ConvTranspose2d): ...@@ -42,7 +42,7 @@ class ConvTranspose2d(torch.nn.ConvTranspose2d):
list(self.output_padding), list(self.output_padding),
) )
] ]
output_shape = [x.shape[0], self.bias.shape[0]] + output_shape output_shape = [x.shape[0], self.out_channels] + output_shape
return _new_empty_tensor(x, output_shape) return _new_empty_tensor(x, output_shape)
def super_forward(self, input, output_size=None): def super_forward(self, input, output_size=None):
......
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