Fix DyDCNv2 RuntimeError (#8485)

the parameter of offset is not set as continuous will trigger the runtime error: offset must be continuous
pull/8799/head
James 3 years ago committed by ZwwWayne
parent d929c630a0
commit e9ca9d314f
  1. 2
      mmdet/models/necks/dyhead.py

@ -44,7 +44,7 @@ class DyDCNv2(nn.Module):
def forward(self, x, offset, mask):
"""Forward function."""
x = self.conv(x.contiguous(), offset, mask)
x = self.conv(x.contiguous(), offset.contiguous(), mask)
if self.with_norm:
x = self.norm(x)
return x

Loading…
Cancel
Save