mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Fix default_cfg test for mobilenet_100

Browse Source
This commit is contained in:
Ross Wightman 2024-07-09 12:52:24 -07:00
parent d5afe106dc
commit 644abf9588
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/models/efficientnet.py
View File

@ -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)