Update head metadata for effformerv2

rwightman · rwightman · commit 13acac8c5e26 · 2023-02-04T23:11:51.000-08:00
diff --git a/timm/models/efficientformer_v2.py b/timm/models/efficientformer_v2.py
@@ -653,7 +653,7 @@ def _cfg(url='', **kwargs):
         'num_classes': 1000, 'input_size': (3, 224, 224), 'pool_size': None, 'fixed_input_size': True,
         'crop_pct': .95, 'interpolation': 'bicubic',
         'mean': IMAGENET_DEFAULT_MEAN, 'std': IMAGENET_DEFAULT_STD,
-        'classifier': 'head', 'first_conv': 'stem.conv1.conv',
+        'classifier': ('head', 'head_dist'), 'first_conv': 'stem.conv1.conv',
         **kwargs
     }
 

Original file line number	Diff line number	Diff line change
`@@ -653,7 +653,7 @@ def _cfg(url='', **kwargs):`
`653`	`653`	`'num_classes': 1000, 'input_size': (3, 224, 224), 'pool_size': None, 'fixed_input_size': True,`
`654`	`654`	`'crop_pct': .95, 'interpolation': 'bicubic',`
`655`	`655`	`'mean': IMAGENET_DEFAULT_MEAN, 'std': IMAGENET_DEFAULT_STD,`
`656`		`- 'classifier': 'head', 'first_conv': 'stem.conv1.conv',`
	`656`	`+ 'classifier': ('head', 'head_dist'), 'first_conv': 'stem.conv1.conv',`
`657`	`657`	`**kwargs`
`658`	`658`	`}`
`659`	`659`