Add D-FINE Model into Transformers (#36261)
* copy the last changes from broken PR * small format * some fixes and refactoring after review * format * add config attr for loss * some fixes and refactoring * fix copies * fix style * add test for d-fine resnet * fix decoder layer prop * fix dummies * format init * remove extra print * refactor modeling, move resnet into separate folder * fix resnet config * change resnet on hgnet_v2, add clamp into decoder * fix init * fix config doc * fix init * fix dummies * fix config docs * fix hgnet_v2 config typo * format modular * add image classification for hgnet, some refactoring * format tests * fix dummies * fix init * fix style * fix init for hgnet v2 * fix index.md, add init rnage for hgnet * fix conversion * add missing attr to encoder * add loss for d-fine, add additional output for rt-detr decoder * tests and docs fixes * fix rt_detr v2 conversion * some fixes for loos and decoder output * some fixes for loss * small fix for converted modeling * add n model config, some todo comments for modular * convert script adjustments and fixes, small refact * remove extra output for rt_detr * make some outputs optionsl, fix conversion * some posr merge fixes * small fix * last field fix * fix not split for hgnet_v2 * disable parallelism test for hgnet_v2 image classification * skip multi gpu for d-fine * adjust after merge init * remove extra comment * fix repo name references * small fixes for tests * Fix checkpoint path * Fix consistency * Fixing docs --------- Co-authored-by: Pavel Iakubovskii <qubvel@gmail.com>
This commit is contained in:
committed by
GitHub
parent
4602059aae
commit
4abeb50f6e
@@ -183,6 +183,22 @@ SPECIAL_CASES_TO_ALLOW = {
|
||||
"giou_loss_coefficient",
|
||||
"mask_loss_coefficient",
|
||||
],
|
||||
"DFineConfig": [
|
||||
"eos_coefficient",
|
||||
"focal_loss_alpha",
|
||||
"focal_loss_gamma",
|
||||
"matcher_alpha",
|
||||
"matcher_bbox_cost",
|
||||
"matcher_class_cost",
|
||||
"matcher_gamma",
|
||||
"matcher_giou_cost",
|
||||
"use_focal_loss",
|
||||
"weight_loss_bbox",
|
||||
"weight_loss_giou",
|
||||
"weight_loss_vfl",
|
||||
"weight_loss_fgl",
|
||||
"weight_loss_ddf",
|
||||
],
|
||||
"GroundingDinoConfig": [
|
||||
"bbox_cost",
|
||||
"bbox_loss_coefficient",
|
||||
|
||||
Reference in New Issue
Block a user