提交 8ceceef8 编写于 作者: R Ross Wightman

Densenet should default to bicubic interpolation, update model links to 'cpu fix' from long ago

上级 a4516fe0
......@@ -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'),
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册