Fix default_cfg test for mobilenet_100

huggingface · Jul 9, 2024 · 644abf9 · 644abf9
1 parent d5afe10
commit 644abf9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/timm/models/efficientnet.py b/timm/models/efficientnet.py
@@ -131,7 +131,7 @@ def __init__(
  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)