NielsRogge/transformers

Pull Requests Commits
Apply suggestions from code review
NielsRogge committed 3 years ago
80d2e8d9
Fix training integration test
NielsRogge committed 3 years ago
edf09877
Delete scripts and add integration test for training
NielsRogge committed 3 years ago
c911c88f
Add integration test, general fixes
NielsRogge committed 3 years ago
0519afb7
Remove img_metas default value
NielsRogge committed 3 years ago
17ba740e
Activate mask loss
NielsRogge committed 3 years ago
8fb1817d
Deactive mask loss
NielsRogge committed 3 years ago
2a0ae2c5
Remove checké
NielsRogge committed 3 years ago
5c8aaa03
Add cpu method
NielsRogge committed 3 years ago
90741b05
Add losses to output
NielsRogge committed 3 years ago
c33be01d
Improve docstrings
NielsRogge committed 3 years ago
03d349e1
Add loss computation of mask head
NielsRogge committed 3 years ago
40e80f3e
Add loss calculation of mask head
NielsRogge committed 3 years ago
f8541aa6
Add forward_train to roi head
NielsRogge committed 3 years ago
8d373fba
More improvements
NielsRogge committed 3 years ago
65166478
Add training support for RPN
NielsRogge committed 3 years ago
f32aa4d8
Improve forward of RPN
NielsRogge committed 3 years ago
8b5c1161
Add sampler
NielsRogge committed 3 years ago
880c7586
Add Assigner and BboxOverlaps classes
NielsRogge committed 3 years ago
548ca956
Verify mask head
NielsRogge committed 3 years ago
04c9bbda
Add mask head
NielsRogge committed 3 years ago
7392f262
Make bounding box results match original implementation
NielsRogge committed 3 years ago
71b05e1b
Add initial draft of RoI head
NielsRogge committed 3 years ago
0dea9cdc
Use torchvision's nms implementation
NielsRogge committed 3 years ago
4eac5d1b
More improvements
NielsRogge committed 3 years ago
e7ea1668
More improvements
NielsRogge committed 3 years ago
533eb39c
More fixes
NielsRogge committed 3 years ago
dcf92f6d
More fixes
NielsRogge committed 3 years ago
f2e9fba7
Fix conversion without layernorm
NielsRogge committed 3 years ago
f076eabd
More improvements
NielsRogge committed 3 years ago
90ea55b7
Terms Of Use Privacy Policy Refund Policy Impressum