We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a5a2ad2 commit 5ee0676Copy full SHA for 5ee0676
timm/models/mobilenetv3.py
@@ -134,7 +134,7 @@ def __init__(
134
self.norm_head = nn.Identity()
135
self.act2 = act_layer(inplace=True)
136
self.flatten = nn.Flatten(1) if global_pool else nn.Identity() # don't flatten if pooling disabled
137
- self.classifier = Linear(self.num_features, num_classes) if num_classes > 0 else nn.Identity()
+ self.classifier = Linear(self.head_hidden_size, num_classes) if num_classes > 0 else nn.Identity()
138
139
efficientnet_init_weights(self)
140
0 commit comments