From 8682528096f172ff8e0157e14845da748ce94bb5 Mon Sep 17 00:00:00 2001
From: Ross Wightman <rwightman@gmail.com>
Date: Sat, 4 Feb 2023 23:02:02 -0800
Subject: [PATCH] Add first conv metadata for efficientformer_v2

---
 timm/models/efficientformer_v2.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/timm/models/efficientformer_v2.py b/timm/models/efficientformer_v2.py
index 737e314a..2f76fb1e 100644
--- 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',
+        'classifier': 'head', 'first_conv': 'stem.conv1.conv',
         **kwargs
     }