Skip to content

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.

3 participants