transformers
048f599f
- Fix RT-DETR weights initialization (#31724)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
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
References
#31724 - Fix RT-DETR weights initialization
Author
qubvel
Parents
b9752161
Loading