Skip to content

GitLab

Unverified Commit b63d880a authored by Wang Xinjiang's avatar Wang Xinjiang Committed by GitHub
Browse files

Add sleep(2) during epoch transition (#306)

parent ff89d366
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
mmcv/runner/runner.py
View file @ b63d880a
...@@ -275,6 +275,7 @@ class Runner(object): ...@@ -275,6 +275,7 @@ class Runner(object):
self.data_loader = data_loader self.data_loader = data_loader
self.call_hook('before_train_epoch') self.call_hook('before_train_epoch')
time.sleep(2) # Prevent possible deadlock during epoch transition
for i, data_batch in enumerate(data_loader): for i, data_batch in enumerate(data_loader):
self._inner_iter = i self._inner_iter = i
self.call_hook('before_train_iter') self.call_hook('before_train_iter')
...@@ -297,7 +298,7 @@ class Runner(object): ...@@ -297,7 +298,7 @@ class Runner(object):
self.mode = 'val' self.mode = 'val'
self.data_loader = data_loader self.data_loader = data_loader
self.call_hook('before_val_epoch') self.call_hook('before_val_epoch')
time.sleep(2) # Prevent possible deadlock during epoch transition
for i, data_batch in enumerate(data_loader): for i, data_batch in enumerate(data_loader):
self._inner_iter = i self._inner_iter = i
self.call_hook('before_val_iter') self.call_hook('before_val_iter')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment