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

Add missing device info on RetinaNet #3651

Merged
merged 2 commits into from
Apr 8, 2021

Conversation

datumbox
Copy link
Contributor

@datumbox datumbox commented Apr 8, 2021

RetinaNet does not define device info on the index vector when the boxes are empty. This fixes the bug.

Copy link
Member

@fmassa fmassa left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks!

@fmassa fmassa merged commit 07fb8ba into pytorch:master Apr 8, 2021
@datumbox datumbox deleted the bugfix/retinanet_device_def branch April 8, 2021 14:58
facebook-github-bot pushed a commit that referenced this pull request Apr 13, 2021
Reviewed By: NicolasHug

Differential Revision: D27706953

fbshipit-source-id: c3810183d8acb7feec6868e28705ee25806acc27
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants