transformers
ee298a16 - Align backbone stage selection with out_indices & out_features (#27606)

Commit
2 years ago
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
Author
Parents
Loading