Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "dc7c49e4e419ef0888647873b0fb2e233fea6dc2"
Commit cc81d4b5 authored by RangiLyu's avatar RangiLyu Committed by Wenwei Zhang
Browse files

[Enhance] prevent deadlock in

parent 8afb7c29
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
mmcv/runner/iter_based_runner.py
View file @ cc81d4b5
...@@ -34,6 +34,7 @@ class IterLoader: ...@@ -34,6 +34,7 @@ class IterLoader:
self._epoch += 1 self._epoch += 1
if hasattr(self._dataloader.sampler, 'set_epoch'): if hasattr(self._dataloader.sampler, 'set_epoch'):
self._dataloader.sampler.set_epoch(self._epoch) self._dataloader.sampler.set_epoch(self._epoch)
time.sleep(2) # Prevent possible deadlock during epoch transition
self.iter_loader = iter(self._dataloader) self.iter_loader = iter(self._dataloader)
data = next(self.iter_loader) data = next(self.iter_loader)
......
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