Update vanillanet.py

huawei-noah · May 25, 2023 · 2a02a70 · 2a02a70
1 parent f8385cf
commit 2a02a70
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/vanillanet.py b/models/vanillanet.py
@@ -30,7 +30,7 @@ def forward(self, x):
         else:
             return self.bn(torch.nn.functional.conv2d(
                 super(activation, self).forward(x),
-                self.weight, padding=(self.act_num*2 + 1)//2, groups=self.dim))
+                self.weight, padding=self.act_num, groups=self.dim))
 
     def _fuse_bn_tensor(self, weight, bn):
         kernel = weight