turn off persistent_workers while preparing nncf

openvinotoolkit · Jun 8, 2023 · 1f9e4d4 · 1f9e4d4
1 parent 41c9b9b
commit 1f9e4d4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/otx/algorithms/classification/adapters/mmcls/nncf/builder.py b/otx/algorithms/classification/adapters/mmcls/nncf/builder.py
@@ -83,6 +83,7 @@ def build_nncf_classifier(  # pylint: disable=too-many-locals,too-many-statement
             subset="val",
             dataloader_builder=mmcls_build_dataloader,
             distributed=distributed,
+            persistent_workers=False,
         )
 
         # This data and state dict will be used to build NNCF graph later
@@ -110,6 +111,7 @@ def build_nncf_classifier(  # pylint: disable=too-many-locals,too-many-statement
                 subset="val",
                 dataloader_builder=mmcls_build_dataloader,
                 distributed=distributed,
+                persistent_workers=False,
             )
 
         model_eval_fn = partial(

diff --git a/otx/algorithms/detection/adapters/mmdet/nncf/builder.py b/otx/algorithms/detection/adapters/mmdet/nncf/builder.py
@@ -94,6 +94,7 @@ def build_nncf_detector(  # pylint: disable=too-many-locals,too-many-statements
             subset="val",
             dataloader_builder=mmdet_build_dataloader,
             distributed=distributed,
+            persistent_workers=False,
         )
 
         # This data and state dict will be used to build NNCF graph later
@@ -121,6 +122,7 @@ def build_nncf_detector(  # pylint: disable=too-many-locals,too-many-statements
                 subset="val",
                 dataloader_builder=mmdet_build_dataloader,
                 distributed=distributed,
+                persistent_workers=False,
             )
 
         model_eval_fn = partial(