Skip to content

GitLab

Commit 8b4de1cb authored by Michael Jungo's avatar Michael Jungo Committed by Francisco Massa
Browse files

Fix DeformConv2d creation with integer kernel size (#1637)

parent cec7ea72
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
torchvision/ops/deform_conv.py
View file @ 8b4de1cb
...@@ -79,7 +79,8 @@ class DeformConv2d(nn.Module): ...@@ -79,7 +79,8 @@ class DeformConv2d(nn.Module):
self.groups = groups self.groups = groups
self.offset_groups = offset_groups self.offset_groups = offset_groups
self.weight = Parameter(torch.empty(out_channels, in_channels // groups, kernel_size[0], kernel_size[1])) self.weight = Parameter(torch.empty(out_channels, in_channels // groups,
self.kernel_size[0], self.kernel_size[1]))
if bias: if bias:
self.bias = Parameter(torch.empty(out_channels)) self.bias = Parameter(torch.empty(out_channels))
......
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