diff --git a/timm/models/vovnet.py b/timm/models/vovnet.py index bedff10c..70dbac12 100644 --- a/timm/models/vovnet.py +++ b/timm/models/vovnet.py @@ -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