Skip to content

Commit e79eb08

Browse files
authored
[OTE] Comment out time.sleep(2) in EpochRunnerWithCancel (#1174)
* Remove time.sleep(2) in EpochRunnerWithCancel Co-authored-by: eunwoosh <eunwoo.shin@intel.com>
1 parent 5dd1c47 commit e79eb08

File tree

3 files changed

+6
-3
lines changed
  • external
    • mmdetection/detection_tasks/extension/utils
    • mmsegmentation/segmentation_tasks/extension/utils
    • model-preparation-algorithm/mpa_tasks/utils

3 files changed

+6
-3
lines changed

external/mmdetection/detection_tasks/extension/utils/runner.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,8 @@ def train(self, data_loader: DataLoader, **kwargs):
5757
self.data_loader = data_loader
5858
self._max_iters = self._max_epochs * len(self.data_loader)
5959
self.call_hook('before_train_epoch')
60-
time.sleep(2) # Prevent possible deadlock during epoch transition
60+
# TODO: uncomment below line or resolve root cause of deadlock issue if multi-GPUs need to be supported.
61+
# time.sleep(2) # Prevent possible multi-gpu deadlock during epoch transition
6162
for i, data_batch in enumerate(self.data_loader):
6263
self._inner_iter = i
6364
self.call_hook('before_train_iter')

external/mmsegmentation/segmentation_tasks/extension/utils/runner.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,8 @@ def train(self, data_loader: DataLoader, **kwargs):
6161
self.data_loader = data_loader
6262
self._max_iters = self._max_epochs * len(self.data_loader)
6363
self.call_hook('before_train_epoch')
64-
time.sleep(2) # Prevent possible deadlock during epoch transition
64+
# TODO: uncomment below line or resolve root cause of deadlock issue if multi-GPUs need to be supported.
65+
# time.sleep(2) # Prevent possible multi-gpu deadlock during epoch transition
6566
for i, data_batch in enumerate(self.data_loader):
6667
self._inner_iter = i
6768
self.call_hook('before_train_iter')

external/model-preparation-algorithm/mpa_tasks/utils/runner.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,8 @@ def train(self, data_loader: DataLoader, **kwargs):
5050
self.data_loader = data_loader
5151
self._max_iters = self._max_epochs * len(self.data_loader)
5252
self.call_hook('before_train_epoch')
53-
time.sleep(2) # Prevent possible deadlock during epoch transition
53+
# TODO: uncomment below line or resolve root cause of deadlock issue if multi-GPUs need to be supported.
54+
# time.sleep(2) # Prevent possible multi-gpu deadlock during epoch transition
5455
for i, data_batch in enumerate(self.data_loader):
5556
self._inner_iter = i
5657
self.call_hook('before_train_iter')

0 commit comments

Comments
 (0)