Skip to content

Fix regression on Detection training script #5985

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

Merged
merged 1 commit into from
May 10, 2022

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