d-li14/PSConv

I wonder why the code psconv.py are not consistent with the description in the paper?

wuzhenyubuaa opened this issue · 0 comments

x1, x2 = x.chunk(2, dim=1)
x_shift = self.gwconv_shift(torch.cat((x2, x1), dim=1))
return self.gwconv(x) + self.conv(x) + x_shift

I think the code is simply add with different dilation.