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

[cudnn] set False to cudnn benchmark by default #1063

Merged
merged 1 commit into from
Jun 3, 2022

Conversation

FrankLeeeee
Copy link
Contributor

Fixed #1058 , set cudnn benchmark to false so that it will not repeatedly run benchmarking when inputs are of varying sizes.

If the user wants to turn on cudnn, they can have cudnn_benchmark = True in config.py.

@BoxiangW BoxiangW merged commit 3d10be3 into hpcaitech:main Jun 3, 2022
@FrankLeeeee FrankLeeeee deleted the hotfix/cudnn branch June 13, 2022 02:08
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.

[BUG]: Process significantly slowed down and huge memory taken when dealling with images with different sizes
2 participants