Fix typo in Densenet default resolutions

pull/19/head
Ross Wightman 6 years ago
parent da52fcf78a
commit 949b7a81c4

@ -19,7 +19,7 @@ __all__ = ['DenseNet']
def _cfg(url=''): def _cfg(url=''):
return { return {
'url': url, 'num_classes': 1000, 'input_size': (3, 224, 244), 'pool_size': (7, 7), 'url': url, 'num_classes': 1000, 'input_size': (3, 224, 224), 'pool_size': (7, 7),
'crop_pct': 0.875, 'interpolation': 'bicubic', 'crop_pct': 0.875, 'interpolation': 'bicubic',
'mean': IMAGENET_DEFAULT_MEAN, 'std': IMAGENET_DEFAULT_STD, 'mean': IMAGENET_DEFAULT_MEAN, 'std': IMAGENET_DEFAULT_STD,
'first_conv': 'features.conv0', 'classifier': 'classifier', 'first_conv': 'features.conv0', 'classifier': 'classifier',

Loading…
Cancel
Save