transformers/tests/models/dpt
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
test_image_processing_dpt.py Add Swinv2 backbone (#27742) 2023-12-22 11:12:56 +00:00
test_modeling_dpt_auto_backbone.py Add Swinv2 backbone (#27742) 2023-12-22 11:12:56 +00:00
test_modeling_dpt_hybrid.py Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01:00
test_modeling_dpt.py Update forward signature test for vision models (#27681) 2023-11-27 15:48:17 +01:00