NielsRogge/transformers

Pull Requests Commits

Add loss computation of mask head

NielsRogge committed 2 years ago

e3af22e4

Add loss calculation of mask head

NielsRogge committed 2 years ago

05a9e7c1

Add forward_train to roi head

NielsRogge committed 2 years ago

679780f8

More improvements

NielsRogge committed 2 years ago

951ebb1d

Add training support for RPN

NielsRogge committed 2 years ago

274493ff

Improve forward of RPN

NielsRogge committed 2 years ago

67853d19

NielsRogge committed 2 years ago

85af48a0

Add Assigner and BboxOverlaps classes

NielsRogge committed 2 years ago

797b3fb5

Verify mask head

NielsRogge committed 2 years ago

ef2f53f3

NielsRogge committed 2 years ago

f7240d22

Make bounding box results match original implementation

NielsRogge committed 2 years ago

e9adfb6c

Add initial draft of RoI head

NielsRogge committed 2 years ago

d6fd3d5d

Use torchvision's nms implementation

NielsRogge committed 2 years ago

044576f8

More improvements

NielsRogge committed 2 years ago

0783d15c

More improvements

NielsRogge committed 2 years ago

f1e9a6d8

NielsRogge committed 2 years ago

662c439b

NielsRogge committed 2 years ago

867e3ae1

Fix conversion without layernorm

NielsRogge committed 2 years ago

ba797243

More improvements

NielsRogge committed 2 years ago

135cdd66

NielsRogge committed 2 years ago

40e69233

Fix conversion script for backbone

NielsRogge committed 2 years ago

7a1b4956

NielsRogge committed 2 years ago

81333fa2

vilt_model (#22930)

sushmanthreddy committed 2 years ago

Verified d04ec99b

Feature to convert videomae huge and small finetuned on kinetics and ssv2 added to the videomae to pytorch converter (#22788)

sandstorm12 committed 2 years ago

Verified 4d10de55

Small sam patch (#22920)

ArthurZucker committed 2 years ago

Verified 7579a52b

Fix a minor bug in CI slack report (#22906)

ydshieh committed 2 years ago

Verified 5166c30e

tests: Fix flaky test for NLLB-MoE (#22880)

Connor Henderson committed 2 years ago

Verified b950c385

ddp fixes for training (#22874)

winglian committed 2 years ago

Verified d00997e6

[CI] clap patch fusion test values (#22922)

ArthurZucker committed 2 years ago

Verified eddf9eec

Hardcode GELU as the intermediate activation for ESM (#22892)

Rocketknight1 committed 2 years ago

Verified 5600e6f3

Newer Older