[NAT, DiNAT] Add backbone class (#20654)

* Add first draft

* Add out_features attribute to config

* Add corresponding test

* Add Dinat backbone

* Add BackboneMixin

* Add Backbone mixin, improve tests

* Fix embeddings

* Fix bug

* Improve backbones

* Fix Nat backbone tests

* Fix Dinat backbone tests

* Apply suggestions

Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
This commit is contained in:
NielsRogge
2022-12-13 17:06:59 +01:00
committed by GitHub
parent 30d8919ab1
commit 6ef42587ae
12 changed files with 449 additions and 60 deletions

View File

@@ -677,6 +677,8 @@ SHOULD_HAVE_THEIR_OWN_PAGE = [
"MaskFormerSwinBackbone",
"ResNetBackbone",
"AutoBackbone",
"DinatBackbone",
"NatBackbone",
"MaskFormerSwinConfig",
"MaskFormerSwinModel",
]