transformers
LwDetrImageLoss: Fix dtype casting to prevent crash when using amp on cuda device
#44886
Merged
Merged
LwDetrImageLoss: Fix dtype casting to prevent crash when using amp on cuda device #44886
yonigozlan
merged
5 commits into
huggingface:main
from
m-matthias:fix_lwdetrloss_amp_cuda_crash
LwDetrImageLoss: Fix dtype casting to prevent crash when using amp on…
fb8b4322
fix PR review comments
6bf8e86d
Merge branch 'main' into fix_lwdetrloss_amp_cuda_crash
be092ecd
add comments and apply similar fix rt_detr
2b71a82c
yonigozlan
approved these changes
on 2026-03-24
Merge branch 'main' into fix_lwdetrloss_amp_cuda_crash
29a72a5a
Login to write a write a comment.
Login via GitHub