tests/models/bit/test_modeling_bit.py · 9a6c6ef97fa5df4b1fb8dbc9e8c10ee3a9ed7e2a · chenpangpang / transformers

[Backbones] Improve out features (#20675) · 9a6c6ef9

NielsRogge authored Dec 09, 2022



* Improve ResNet backbone

* Improve Bit backbone

* Improve docstrings

* Fix default stage

* Apply suggestions from code review
Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>

9a6c6ef9

test_modeling_bit.py 11.6 KB

Replace test_modeling_bit.py