From c870046a4a194a60fa626dfd3c650143ff44887a Mon Sep 17 00:00:00 2001 From: liukuikun <24622904+Harold-lkk@users.noreply.github.com> Date: Fri, 3 Feb 2023 18:57:12 +0800 Subject: [PATCH] [Fix] change cudnn benchmark to false (#1705) --- configs/kie/_base_/default_runtime.py | 2 +- configs/textdet/_base_/default_runtime.py | 2 +- configs/textrecog/_base_/default_runtime.py | 2 +- projects/ABCNet/config/_base_/default_runtime.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/configs/kie/_base_/default_runtime.py b/configs/kie/_base_/default_runtime.py index f38e39ac..bcc5b3fa 100644 --- a/configs/kie/_base_/default_runtime.py +++ b/configs/kie/_base_/default_runtime.py @@ -1,6 +1,6 @@ default_scope = 'mmocr' env_cfg = dict( - cudnn_benchmark=True, + cudnn_benchmark=False, mp_cfg=dict(mp_start_method='fork', opencv_num_threads=0), dist_cfg=dict(backend='nccl'), ) diff --git a/configs/textdet/_base_/default_runtime.py b/configs/textdet/_base_/default_runtime.py index 51074fb3..81480273 100644 --- a/configs/textdet/_base_/default_runtime.py +++ b/configs/textdet/_base_/default_runtime.py @@ -1,6 +1,6 @@ default_scope = 'mmocr' env_cfg = dict( - cudnn_benchmark=True, + cudnn_benchmark=False, mp_cfg=dict(mp_start_method='fork', opencv_num_threads=0), dist_cfg=dict(backend='nccl'), ) diff --git a/configs/textrecog/_base_/default_runtime.py b/configs/textrecog/_base_/default_runtime.py index 6564fcb8..3dcd1051 100644 --- a/configs/textrecog/_base_/default_runtime.py +++ b/configs/textrecog/_base_/default_runtime.py @@ -1,6 +1,6 @@ default_scope = 'mmocr' env_cfg = dict( - cudnn_benchmark=True, + cudnn_benchmark=False, mp_cfg=dict(mp_start_method='fork', opencv_num_threads=0), dist_cfg=dict(backend='nccl'), ) diff --git a/projects/ABCNet/config/_base_/default_runtime.py b/projects/ABCNet/config/_base_/default_runtime.py index 8a1f12a3..4b9b72c5 100644 --- a/projects/ABCNet/config/_base_/default_runtime.py +++ b/projects/ABCNet/config/_base_/default_runtime.py @@ -1,6 +1,6 @@ default_scope = 'mmocr' env_cfg = dict( - cudnn_benchmark=True, + cudnn_benchmark=False, mp_cfg=dict(mp_start_method='fork', opencv_num_threads=0), dist_cfg=dict(backend='nccl'), )