Make features_only=True work with mnv5 & enc, uses forward_intermediates()

Author: rwightman

timm/models/mobilenetv5.py

@@ -554,6 +554,8 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
 
 
 def _create_mnv5_encoder(variant: str, pretrained: bool = False, **kwargs) -> MobileNetV5Encoder:
+    out_indices = kwargs.pop('out_indices', (0, 1, 2, 3, 4))
+    feature_cfg = dict(out_indices=out_indices, feature_cls='getter')
     kwargs_filter = (
         'num_classes',
         'num_features',
@@ -567,18 +569,22 @@ def _create_mnv5_encoder(variant: str, pretrained: bool = False, **kwargs) -> Mo
         variant,
         pretrained,
         pretrained_strict=False,
+        feature_cfg=feature_cfg,
         kwargs_filter=kwargs_filter,
         **kwargs,
     )
     return model
 
 
 def _create_mnv5(variant: str, pretrained: bool = False, **kwargs) -> MobileNetV5Encoder:
+    out_indices = kwargs.pop('out_indices', (0, 1, 2, 3, 4))
+    feature_cfg = dict(out_indices=out_indices, feature_cls='getter')
     model = build_model_with_cfg(
         MobileNetV5,
         variant,
         pretrained,
         pretrained_strict=False,
+        feature_cfg=feature_cfg,
         **kwargs,
     )
     return model