From c559a6a0713517b8abfbd3c53ec8eb7591530295 Mon Sep 17 00:00:00 2001 From: Jay Young Date: Mon, 3 Aug 2020 15:58:05 +0800 Subject: [PATCH] fix `NameError: name '_remove_worker_pids' is not defined` fix `NameError: name '_remove_worker_pids' is not defined` --- ofa/imagenet_codebase/data_providers/my_data_loader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ofa/imagenet_codebase/data_providers/my_data_loader.py b/ofa/imagenet_codebase/data_providers/my_data_loader.py index 006703e7..300be981 100644 --- a/ofa/imagenet_codebase/data_providers/my_data_loader.py +++ b/ofa/imagenet_codebase/data_providers/my_data_loader.py @@ -6,7 +6,7 @@ import random import torch import torch.multiprocessing as multiprocessing -from torch._C import _set_worker_signal_handlers +from torch._C import _set_worker_signal_handlers, _remove_worker_pids from torch.utils.data import _utils from torch.utils.data import SequentialSampler, RandomSampler, BatchSampler import signal