From 8ceceef889f69a30ffe5c8ef45140a8492a38ddb Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Mon, 20 May 2019 11:45:29 -0700 Subject: [PATCH] Densenet should default to bicubic interpolation, update model links to 'cpu fix' from long ago --- models/densenet.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/models/densenet.py b/models/densenet.py index 8a2e4369..31ce5684 100644 --- a/models/densenet.py +++ b/models/densenet.py @@ -18,17 +18,17 @@ __all__ = ['DenseNet', 'densenet121', 'densenet169', 'densenet201', 'densenet161 def _cfg(url=''): return { 'url': url, 'num_classes': 1000, 'input_size': (3, 224, 244), 'pool_size': (7, 7), - 'crop_pct': 0.875, 'interpolation': 'bilinear', + 'crop_pct': 0.875, 'interpolation': 'bicubic', 'mean': IMAGENET_DEFAULT_MEAN, 'std': IMAGENET_DEFAULT_STD, 'first_conv': 'features.conv0', 'classifier': 'classifier', } default_cfgs = { - 'densenet121': _cfg(url='https://download.pytorch.org/models/densenet121-241335ed.pth'), - 'densenet169': _cfg(url='https://download.pytorch.org/models/densenet169-6f0f7f60.pth'), - 'densenet201': _cfg(url='https://download.pytorch.org/models/densenet201-4c113574.pth'), - 'densenet161': _cfg(url='https://download.pytorch.org/models/densenet161-17b70270.pth'), + 'densenet121': _cfg(url='https://download.pytorch.org/models/densenet121-a639ec97.pth'), + 'densenet169': _cfg(url='https://download.pytorch.org/models/densenet169-b2777c0a.pth'), + 'densenet201': _cfg(url='https://download.pytorch.org/models/densenet201-c1103571.pth'), + 'densenet161': _cfg(url='https://download.pytorch.org/models/densenet161-8d451a50.pth'), }