Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • amyeroberts's avatar
    Align backbone stage selection with out_indices & out_features (#27606) · ee298a16
    amyeroberts authored 
    * 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
    ee298a16
test_backbone_utils.py 5.52 KB