From 79f615639e2cde7d686ed28e4fbcff38c0ce9c8f Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Sat, 13 Apr 2019 14:52:21 -0700 Subject: [PATCH] Add pretrained weights for seresnet18 --- models/senet.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/models/senet.py b/models/senet.py index c8be0769..7676ae99 100644 --- a/models/senet.py +++ b/models/senet.py @@ -29,6 +29,7 @@ def _cfg(url='', **kwargs): 'crop_pct': 0.875, 'interpolation': 'bilinear', 'mean': IMAGENET_DEFAULT_MEAN, 'std': IMAGENET_DEFAULT_STD, 'first_conv': 'layer0.conv1', 'classifier': 'last_linear', + **kwargs } @@ -36,7 +37,8 @@ default_cfgs = { 'senet154': _cfg(url='http://data.lip6.fr/cadene/pretrainedmodels/senet154-c7b49a05.pth'), 'seresnet18': - _cfg(url=''), + _cfg(url='https://www.dropbox.com/s/3o3nd8mfhxod7rq/seresnet18-4bb0ce65.pth?dl=1', + interpolation='bicubic'), 'seresnet34': _cfg(url='https://www.dropbox.com/s/q31ccy22aq0fju7/seresnet34-a4004e63.pth?dl=1'), 'seresnet50':