Skip to content

Conversation

datumbox
Copy link
Contributor

Detection models use variable input sizes and thus shouldn't set torch.backends.cudnn.benchmark=True. Doing so causes a significant speed regression.

Copy link
Contributor

@YosuaMichael YosuaMichael left a comment

Choose a reason for hiding this comment

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

LGTM

@datumbox datumbox merged commit 3ec4b94 into pytorch:main May 10, 2022
@datumbox datumbox deleted the references/deterministic_detection branch May 10, 2022 17:33
facebook-github-bot pushed a commit that referenced this pull request May 20, 2022
Reviewed By: datumbox

Differential Revision: D36413361

fbshipit-source-id: 8e61cae70c97255ebc143991eaad5a07633d317c
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.

2 participants