Skip to content

Commit 77723c7

Browse files
authored
Add USAM module
1 parent 1941c76 commit 77723c7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,8 @@ def __init__(self, kernel_size=3, padding=1, polish=True):
4646

4747
def __call__(self, x):
4848
fmap = x.sum(1, keepdim=True)
49-
x1 = F.conv2d(fmap, self.weight, padding=self.pad)
50-
x2 = F.conv2d(fmap, self.weight2, padding=0)
49+
x1 = nn.functional.conv2d(fmap, self.weight, padding=self.pad)
50+
x2 = nn.functional.conv2d(fmap, self.weight2, padding=0)
5151

5252
att = x2 - x1
5353
att = self.bn(att)

0 commit comments

Comments
 (0)