riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-20 04:58:22 +06:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity
080e14b24c
transformers/tests/models/rt_detr
History
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