From 213d54026683a641695e62df3b425b23537d7278 Mon Sep 17 00:00:00 2001 From: Shivam Singhal Date: Fri, 30 Aug 2024 11:33:25 +0530 Subject: [PATCH] feat(deepface): no optimization --- deepface/models/facial_recognition/Facenet.py | 9 +-------- 1 file changed, 1 insertion(+), 8 deletions(-) diff --git a/deepface/models/facial_recognition/Facenet.py b/deepface/models/facial_recognition/Facenet.py index da67af6..93f980b 100644 --- a/deepface/models/facial_recognition/Facenet.py +++ b/deepface/models/facial_recognition/Facenet.py @@ -1754,14 +1754,7 @@ def load_facenet512d_onnx_model( if torch.cuda.is_available(): logger.info("using onnx GPU for inference") providers = [ - ('CUDAExecutionProvider', { - 'arena_extend_strategy': 'kNextPowerOfTwo', - 'cudnn_conv_algo_search': 'EXHAUSTIVE', - 'do_copy_in_default_stream': True, - 'cudnn_conv_use_max_workspace': '1', - 'cudnn_conv1d_pad_to_nc1d': '1', - 'enable_cuda_graph': '1' - }), + 'CUDAExecutionProvider', 'CPUExecutionProvider', ] else: