From 2a02a70f73c6a3da88c68b00e9bb8052d7ff8d9c Mon Sep 17 00:00:00 2001 From: HantingChen <40243544+HantingChen@users.noreply.github.com> Date: Thu, 25 May 2023 20:54:30 +0800 Subject: [PATCH] Update vanillanet.py --- models/vanillanet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/vanillanet.py b/models/vanillanet.py index 35fbd45..6eef1ed 100644 --- 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