mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Add ESE-VovNet39 weights, 79.32 top-1

Browse Source
This commit is contained in:
Ross Wightman 2020-06-06 22:49:19 -07:00
parent 4ddde1d3a4
commit 13c0a6290e
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
timm/models/vovnet.py
View File

@ -149,7 +149,8 @@ default_cfgs = dict(
ese_vovnet19b_slim_dw=_cfg(url=''),
ese_vovnet19b_dw=_cfg(url=''),
ese_vovnet19b_slim=_cfg(url=''),
ese_vovnet39b=_cfg(url=''),
ese_vovnet39b=_cfg(
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/ese_vovnet39b-f912fe73.pth'),
ese_vovnet57b=_cfg(url=''),
ese_vovnet99b=_cfg(url=''),
eca_vovnet39b=_cfg(url=''),
@ -313,6 +314,8 @@ class VovNet(nn.Module):
elif isinstance(m, nn.BatchNorm2d):
nn.init.constant_(m.weight, 1.)
nn.init.constant_(m.bias, 0.)
elif isinstance(m, nn.Linear):
nn.init.zeros_(m.bias)
def get_classifier(self):
return self.head.fc