Fix typo in Densenet default resolutions

5 years ago · 949b7a81c4
parent da52fcf78a
commit 949b7a81c4
1 changed files with 1 additions and 1 deletions
--- a/timm/models/densenet.py
+++ b/timm/models/densenet.py
@ -19,7 +19,7 @@ __all__ = ['DenseNet']

 def _cfg(url=''):
    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',
        'mean': IMAGENET_DEFAULT_MEAN, 'std': IMAGENET_DEFAULT_STD,
        'first_conv': 'features.conv0', 'classifier': 'classifier',