Add cuda_custom_kernel in DETA #28989
amyeroberts
merged
24 commits into
huggingface:main
from
SangbumChoi:fix_deta
enable graident checkpointing in DetaObjectDetection
769c4c3b
fix missing part in original DETA
2728546c
make style
1448ef84
make fix-copies
33860042
Revert "make fix-copies"
ce901c17
remove fix-copies of DetaDecoder
eb67ff40
enable swin gradient checkpointing
c34d4a9d
fix gradient checkpointing in donut_swin
f455c562
add tests for deta/swin/donut
fb5b0619
Revert "fix gradient checkpointing in donut_swin"
ef1f2395
change supports_gradient_checkpointing pipeline to PreTrainedModel
2278b3d6
Revert "add tests for deta/swin/donut"
d4092fb3
Revert "Revert "fix gradient checkpointing in donut_swin""
06a9f8f9
Simple revert
77caba7d
enable deformable detr gradient checkpointing
d43dcfb0
Merge branch 'main' of https://github.com/SangbumChoi/transformers in…
e2a3d156
add gradient in encoder
536d0bfe
Merge branch 'main' of https://github.com/SangbumChoi/transformers in…
25475d08
add cuda_custom_kernel function in MSDA
081bcf0b
make style and fix input of DetaMSDA
b608dae3
make fix-copies
446c5567
remove n_levels in input of DetaMSDA
c47fe540
minor changes
cf2e3631
refactor custom_cuda_kernel like yoso format
2f5f7a4d
amyeroberts
approved these changes
on 2024-02-15
Login to write a write a comment.
Login via GitHub