transformers
6ef42587 - [NAT, DiNAT] Add backbone class (#20654)

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