-
Notifications
You must be signed in to change notification settings - Fork 8
Open
Description
In Dynamic Semantic Matching Module (DSMM) (line 179):
for i in range(1, B4):
kernel4 = k4[i, :, :, :]
kernel3 = k3[i, :, :, :]
kernel4 = kernel4.view(C4, 1, H4, W4)
kernel3 = kernel3.view(C3, 1, H3, W3)
# DDconv padding 2 4 6
x4_r1 = F.conv2d(x4[i, :, :, :].view(1, C4, x_H4, x_W4), kernel4, stride=1, padding=4, dilation=1, groups=C4)
x4_r2 = F.conv2d(x4[i, :, :, :].view(1, C4, x_H4, x_W4), kernel4, stride=1, padding=8, dilation=2, groups=C4)
x4_r3 = F.conv2d(x4[i, :, :, :].view(1, C4, x_H4, x_W4), kernel4, stride=1, padding=12, dilation=3, groups=C4)
x4_new[i, :, :, :] = x4_r1 + x4_r2 + x4_r3
# DDconv
x3_r1 = F.conv2d(x3[i, :, :, :].view(1, C3, x_H3, x_W3), kernel3, stride=1, padding=4, dilation=1, groups=C3)
x3_r2 = F.conv2d(x3[i, :, :, :].view(1, C3, x_H3, x_W3), kernel3, stride=1, padding=8, dilation=2, groups=C3)
x3_r3 = F.conv2d(x3[i, :, :, :].view(1, C3, x_H3, x_W3), kernel3, stride=1, padding=12, dilation=3, groups=C3)
x3_new[i, :, :, :] = x3_r1 + x3_r2 + x3_r3
range(1, B4) should be range(B4) ?
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels