Skip to content

Commit 13c0a62

Browse files
committed
Add ESE-VovNet39 weights, 79.32 top-1
1 parent 4ddde1d commit 13c0a62

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

timm/models/vovnet.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,8 @@ def _cfg(url=''):
149149
ese_vovnet19b_slim_dw=_cfg(url=''),
150150
ese_vovnet19b_dw=_cfg(url=''),
151151
ese_vovnet19b_slim=_cfg(url=''),
152-
ese_vovnet39b=_cfg(url=''),
152+
ese_vovnet39b=_cfg(
153+
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/ese_vovnet39b-f912fe73.pth'),
153154
ese_vovnet57b=_cfg(url=''),
154155
ese_vovnet99b=_cfg(url=''),
155156
eca_vovnet39b=_cfg(url=''),
@@ -313,6 +314,8 @@ def __init__(self, cfg, in_chans=3, num_classes=1000, global_pool='avg', drop_ra
313314
elif isinstance(m, nn.BatchNorm2d):
314315
nn.init.constant_(m.weight, 1.)
315316
nn.init.constant_(m.bias, 0.)
317+
elif isinstance(m, nn.Linear):
318+
nn.init.zeros_(m.bias)
316319

317320
def get_classifier(self):
318321
return self.head.fc

0 commit comments

Comments
 (0)