From 644abf9588f65406f0c562405b01b4177e8eafdd Mon Sep 17 00:00:00 2001
From: Ross Wightman <rwightman@gmail.com>
Date: Tue, 9 Jul 2024 12:52:24 -0700
Subject: [PATCH] Fix default_cfg test for mobilenet_100

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

diff --git a/timm/models/efficientnet.py b/timm/models/efficientnet.py
index 9a03f7df..289c83f3 100644
--- a/timm/models/efficientnet.py
+++ b/timm/models/efficientnet.py
@@ -131,7 +131,7 @@ class EfficientNet(nn.Module):
         else:
             self.conv_head = nn.Identity()
             self.bn2 = nn.Identity()
-            self.num_features = head_chs
+            self.num_features = self.head_hidden_size = head_chs
 
         self.global_pool, self.classifier = create_classifier(
             self.num_features, self.num_classes, pool_type=global_pool)