Skip to content

Commit c21b216

Browse files
committed
visformer supports spatial feat map, update pool_size in pretrained cfg to match
1 parent 9c11dfd commit c21b216

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

timm/models/visformer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
def _cfg(url='', **kwargs):
2424
return {
2525
'url': url,
26-
'num_classes': 1000, 'input_size': (3, 224, 224), 'pool_size': None,
26+
'num_classes': 1000, 'input_size': (3, 224, 224), 'pool_size': (7, 7),
2727
'crop_pct': .9, 'interpolation': 'bicubic', 'fixed_input_size': True,
2828
'mean': IMAGENET_DEFAULT_MEAN, 'std': IMAGENET_DEFAULT_STD,
2929
'first_conv': 'stem.0', 'classifier': 'head',

0 commit comments

Comments
 (0)