Add ESE-VovNet39 weights, 79.32 top-1

pull/155/head
Ross Wightman 4 years ago
parent 4ddde1d3a4
commit 13c0a6290e

@ -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

Loading…
Cancel
Save