amyeroberts ee298a16a2 Align backbone stage selection with out_indices & out_features (#27606) ...

* Iteratre over out_features instead of stage_names

* Update for all backbones

* Add tests

* Fix

* Align timm backbone behaviour with other backbones

* Fix tests

* Stricter checks on set out_features and out_indices

* Revert back stage selection logic

* Remove out-of-order logic

* Document restriction in docstrings

2023-12-20 18:33:17 +00:00

10 KiB

Raw Blame History

View Raw