Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Fix] Fix argument name for fp32 in DeformableDETRHead #9607

Merged
merged 1 commit into from
Jan 10, 2023

Conversation

nijkah
Copy link
Contributor

@nijkah nijkah commented Jan 10, 2023

Motivation

In DeformableDETRHead, there were wrong argument name for fp32.

Modification

Fix to the appropriate name
Note that torch.cdist in BBoxL1Cost only supports fp32 so the predicted tensors should be converted to fp32.

Additional Information

It will be able to run Deformable DETR with fp16 after open-mmlab/mmcv#2541 is merged.

@nijkah nijkah changed the title [Fix] Fix argument name for fp32 [Fix] Fix argument name for fp32 in DeformableDETRHead Jan 10, 2023
@ZwwWayne ZwwWayne added this to the 2.28.0 milestone Jan 10, 2023
@ZwwWayne ZwwWayne merged commit 721e115 into open-mmlab:dev Jan 10, 2023
@nijkah nijkah deleted the fix/deformp_fp32 branch January 10, 2023 14:12
thmegy pushed a commit to thmegy/mmdetection that referenced this pull request May 5, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants