Pavel Iakubovskii 048f599f35 Fix RT-DETR weights initialization (#31724) ...

* Fix init for rt-detr heads

* Fixup

* Add separate prior_prob value to config for initialization

* Add bbox init

* Change to 1 / num_labels init

* Adjust weights init test

* Fix style for test

2024-07-03 14:29:02 +01:00

..

__init__.py

New model support RTDETR (#29077)

2024-06-21 17:50:08 +01:00

test_image_processing_rt_detr.py

New model support RTDETR (#29077)

2024-06-21 17:50:08 +01:00

test_modeling_rt_detr_resnet.py

New model support RTDETR (#29077)

2024-06-21 17:50:08 +01:00

test_modeling_rt_detr.py

Fix RT-DETR weights initialization (#31724)

2024-07-03 14:29:02 +01:00