Skip to content

Commit 5db1eb6

Browse files
committed
Add defaults
1 parent 9b97753 commit 5db1eb6

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

timm/models/convit.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,9 @@
1919
def _cfg(url='', **kwargs):
2020
return {
2121
'url': url,
22-
'num_classes': 1000, 'input_size': (3, 224, 224),
22+
'num_classes': 1000, 'input_size': (3, 224, 224), 'pool_size': None,
2323
'mean': IMAGENET_DEFAULT_MEAN, 'std': IMAGENET_DEFAULT_STD,
24+
'first_conv': 'patch_embed.proj', 'classifier': 'head',
2425
**kwargs
2526
}
2627

0 commit comments

Comments
 (0)