github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-07 03:55:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(deepface): no optimization

Browse Source
This commit is contained in:
Shivam Singhal 2024-08-30 11:33:25 +05:30
parent e351b1fc2b
commit 213d540266
1 changed files with 1 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
deepface/models/facial_recognition/Facenet.py
View File

@ -1754,14 +1754,7 @@ def load_facenet512d_onnx_model(
if torch.cuda.is_available(): if torch.cuda.is_available():
logger.info("using onnx GPU for inference") logger.info("using onnx GPU for inference")
providers = [ providers = [
('CUDAExecutionProvider', { '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'
}),
'CPUExecutionProvider', 'CPUExecutionProvider',
] ]
else: else: