From 36ecd376dfd618046a557f7b22667a444787ff44 Mon Sep 17 00:00:00 2001
From: Ross Wightman <rwightman@gmail.com>
Date: Fri, 13 Sep 2024 21:14:14 -0700
Subject: [PATCH] Add first_conv for mambaout

---
 timm/models/mambaout.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/timm/models/mambaout.py b/timm/models/mambaout.py
index bda69b1124..91ea0a0914 100644
--- a/timm/models/mambaout.py
+++ b/timm/models/mambaout.py
@@ -314,7 +314,6 @@ def __init__(
             drop_path_rate=0.,
             drop_rate=0.,
             head_fn='default',
-            **kwargs,
     ):
         super().__init__()
         self.num_classes = num_classes
@@ -461,7 +460,8 @@ def _cfg(url='', **kwargs):
         'url': url,
         'num_classes': 1000, 'input_size': (3, 224, 224), 'pool_size': (7, 7),
         'crop_pct': 1.0, 'interpolation': 'bicubic',
-        'mean': IMAGENET_DEFAULT_MEAN, 'std': IMAGENET_DEFAULT_STD, 'classifier': 'head.fc',
+        'mean': IMAGENET_DEFAULT_MEAN, 'std': IMAGENET_DEFAULT_STD,
+        'first_conv': 'stem.conv1', 'classifier': 'head.fc',
         **kwargs
     }