transformers/tests/models/swinv2
NielsRogge c9fb250a25
Add Swinv2 backbone (#27742)
* First draft

* More improvements

* More improvements

* Make all tests pass

* Remove script

* Update image processor

* Address comments

* Use new gradient checkpointing method

* Convert checkpoints, add integration test

* Do not keep aspect ratio for now

* Set keep_aspect_ratio=False for beit, add integration test

* Remove print statement
2023-12-22 11:12:56 +00:00
..
__init__.py Add swin transformer v2 (#17469) 2022-07-27 11:14:47 -04:00
test_modeling_swinv2.py Add Swinv2 backbone (#27742) 2023-12-22 11:12:56 +00:00