Code

Fix RT-DETR weights initialization#31724

Code

Merged

qubvel merged 7 commits intohuggingface:mainhuggingface/transformers:mainfrom qubvel:fix-rt-detr-initqubvel/transformers:fix-rt-detr-initCopy head branch name to clipboardJul 3, 2024

+52-13

Commits

Commits on Jun 28, 2024

• Fix init for rt-detr heads

  qubvel

  committed
  648fabf

  Copy full SHA for 648fabf

• Fixup

  qubvel

  committed
  adbe0e6

  Copy full SHA for adbe0e6

• Add separate prior_prob value to config for initialization

  qubvel

  committed
  0a70c1d

  Copy full SHA for 0a70c1d

• Add bbox init

  qubvel

  committed
  a6df227

  Copy full SHA for a6df227

• Change to 1 / num_labels init

  qubvel

  committed
  5962a73

  Copy full SHA for 5962a73

Commits on Jul 1, 2024

• Adjust weights init test

  qubvel

  committed
  c6e50f8

  Copy full SHA for c6e50f8

• Fix style for test

  qubvel

  committed
  8d200d4

  Copy full SHA for 8d200d4