transformers
6b217c52 - Add AutoBackbone + ResNetBackbone (#20229)

Commit
3 years ago
Add AutoBackbone + ResNetBackbone (#20229) * Add ResNetBackbone * Define channels and strides as property * Remove file * Add test for backbone * Update BackboneOutput class * Remove strides property * Fix docstring * Add backbones to SHOULD_HAVE_THEIR_OWN_PAGE * Fix auto mapping name * Add sanity check for out_features * Set stage names based on depths * Update to tuple Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
Author
Parents
Loading