From f59e0a8dffaf5f608b9c502ddc3c431560cdbdce Mon Sep 17 00:00:00 2001 From: liuqi Date: Sun, 9 Feb 2025 12:59:31 +0800 Subject: [PATCH] fix num wrokers --- msadaptor/torch/utils/data/_utils/worker.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/msadaptor/torch/utils/data/_utils/worker.py b/msadaptor/torch/utils/data/_utils/worker.py index 0b04db23..e7a3f632 100644 --- a/msadaptor/torch/utils/data/_utils/worker.py +++ b/msadaptor/torch/utils/data/_utils/worker.py @@ -15,6 +15,7 @@ import torch from torch._utils import ExceptionWrapper import torch.configs +from mindspore._c_expression import disable_multi_thread from . import HAS_NUMPY, IS_WINDOWS, MP_STATUS_CHECK_INTERVAL, signal_handling @@ -246,6 +247,7 @@ def _worker_loop( # See NOTE [ Data Loader Multiprocessing Shutdown Logic ] for details on the # logic of this function. torch.configs.set_pyboost(use_pyboost) + disable_multi_thread() try: # Initialize C side signal handlers for SIGBUS and SIGSEGV. Python signal # module's handlers are executed after Python returns from C low-level -- Gitee