fix error of min input size

3 years ago · 1d81cd55f5
parent f080f1b8c9
commit 1d81cd55f5
1 changed files with 1 additions and 0 deletions
--- a/timm/models/vision_gnn.py
+++ b/timm/models/vision_gnn.py
@ -22,6 +22,7 @@ def _cfg(url='', **kwargs):
        'crop_pct': .9, 'interpolation': 'bicubic',
        'mean': IMAGENET_DEFAULT_MEAN, 'std': IMAGENET_DEFAULT_STD,
        'first_conv': 'patch_embed.proj', 'classifier': 'head',
+        'min_input_size': (3, 224, 224),
        **kwargs
    }