Use default memory

This commit is contained in:
Nicolas Mowen 2024-10-13 18:13:41 -06:00
parent 71cb9e7fd9
commit 12a0d36602

View File

@ -31,11 +31,7 @@ def get_ort_providers(
for provider in providers:
if provider == "CudaExecutionProvider":
providers.append(provider)
options.append(
{
"arena_extend_strategy": "kSameAsRequested",
}
)
options.append({})
elif provider == "TensorrtExecutionProvider":
# TensorrtExecutionProvider uses too much memory without options to control it
pass