transformers/tests/models/conditional_detr
NielsRogge 90071fe42b
Improve DETR models (#19644)
* Improve DETR models

* Fix Deformable DETR loss and matcher

* Fixup

* Fix integration tests

* Improve variable names

* Apply suggestion

* Fix copies

* Fix DeformableDetrLoss

* Make Conditional DETR copy from Deformable DETR

* Copy from deformable detr's hungarian matcher

* Fix bug
2022-10-18 10:29:14 +02:00
..
__init__.py Add support for conditional detr (#18948) 2022-09-22 09:45:04 +02:00
test_feature_extraction_conditional_detr.py Add support for conditional detr (#18948) 2022-09-22 09:45:04 +02:00
test_modeling_conditional_detr.py Improve DETR models (#19644) 2022-10-18 10:29:14 +02:00