Skip to content

Commit a12b72b

Browse files
committed
Fix missing head_norm arg pop for feature model
1 parent 7fe96e7 commit a12b72b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

timm/models/mobilenetv3.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -385,7 +385,7 @@ def _create_mnv3(variant: str, pretrained: bool = False, **kwargs) -> MobileNetV
385385
if 'feature_cfg' in kwargs or 'feature_cls' in kwargs:
386386
features_mode = 'cfg'
387387
else:
388-
kwargs_filter = ('num_classes', 'num_features', 'head_conv', 'head_bias', 'global_pool')
388+
kwargs_filter = ('num_classes', 'num_features', 'head_conv', 'head_bias', 'head_norm', 'global_pool')
389389
model_cls = MobileNetV3Features
390390
features_mode = 'cls'
391391

0 commit comments

Comments
 (0)