Skip to content

GitLab

Unverified Commit 17e4732c authored by su's avatar su Committed by GitHub
Browse files

Change the epoch runner to use the data_loader from attributes rather than args. (#483)

parent e92f826a
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
mmcv/runner/epoch_based_runner.py
View file @ 17e4732c
...@@ -21,10 +21,10 @@ class EpochBasedRunner(BaseRunner): ...@@ -21,10 +21,10 @@ class EpochBasedRunner(BaseRunner):
self.model.train() self.model.train()
self.mode = 'train' self.mode = 'train'
self.data_loader = data_loader self.data_loader = data_loader
self._max_iters = self._max_epochs * len(data_loader) self._max_iters = self._max_epochs * len(self.data_loader)
self.call_hook('before_train_epoch') self.call_hook('before_train_epoch')
time.sleep(2) # Prevent possible deadlock during epoch transition time.sleep(2) # Prevent possible deadlock during epoch transition
for i, data_batch in enumerate(data_loader): for i, data_batch in enumerate(self.data_loader):
self._inner_iter = i self._inner_iter = i
self.call_hook('before_train_iter') self.call_hook('before_train_iter')
if self.batch_processor is None: if self.batch_processor is None:
...@@ -52,7 +52,7 @@ class EpochBasedRunner(BaseRunner): ...@@ -52,7 +52,7 @@ class EpochBasedRunner(BaseRunner):
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 time.sleep(2) # Prevent possible deadlock during epoch transition
for i, data_batch in enumerate(data_loader): for i, data_batch in enumerate(self.data_loader):
self._inner_iter = i self._inner_iter = i
self.call_hook('before_val_iter') self.call_hook('before_val_iter')
with torch.no_grad(): with torch.no_grad():
......
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