File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed
sdk/ml/azure-ai-ml/azure/ai/ml/_internal/entities Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change 1313
1414import yaml
1515
16- from ..._artifacts ._constants import PROCESSES_PER_CORE
1716from ..._utils ._asset_utils import IgnoreFile , traverse_directory
1817from ..._utils .utils import is_concurrent_component_registration_enabled , is_private_preview_enabled
1918from ...entities ._util import _general_copy
@@ -346,7 +345,10 @@ def _load_artifact_additional_includes(self):
346345 additional_includes_configs = additional_includes_configs .get (ADDITIONAL_INCLUDES_KEY , [])
347346
348347 additional_includes , conflict_files = [], {}
349- num_threads = int (cpu_count ()) * PROCESSES_PER_CORE
348+ # Unlike component registration, artifact downloading is a pure download progress; so we can use
349+ # more threads to speed up the downloading process.
350+ # We use 5 threads per CPU core plus 5 extra threads, and the max number of threads is 64.
351+ num_threads = min (64 , (int (cpu_count ()) * 5 ) + 5 )
350352 if (
351353 len (additional_includes_configs ) > 1
352354 and is_concurrent_component_registration_enabled ()
You can’t perform that action at this time.
0 commit comments