Skip to content

GitLab

Unverified Commit 65a60a3d authored by Rui Xu's avatar Rui Xu Committed by GitHub
Browse files

fix iter-time-hook bug when meeting iter-based runner (#623)

parent c8146cc5
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
mmcv/runner/iter_based_runner.py
View file @ 65a60a3d
...@@ -55,8 +55,8 @@ class IterBasedRunner(BaseRunner): ...@@ -55,8 +55,8 @@ class IterBasedRunner(BaseRunner):
self.mode = 'train' self.mode = 'train'
self.data_loader = data_loader self.data_loader = data_loader
self._epoch = data_loader.epoch self._epoch = data_loader.epoch
self.call_hook('before_train_iter')
data_batch = next(data_loader) data_batch = next(data_loader)
self.call_hook('before_train_iter')
outputs = self.model.train_step(data_batch, self.optimizer, **kwargs) outputs = self.model.train_step(data_batch, self.optimizer, **kwargs)
if not isinstance(outputs, dict): if not isinstance(outputs, dict):
raise TypeError('model.train_step() must return a dict') raise TypeError('model.train_step() must return a dict')
...@@ -71,8 +71,8 @@ class IterBasedRunner(BaseRunner): ...@@ -71,8 +71,8 @@ class IterBasedRunner(BaseRunner):
self.model.eval() self.model.eval()
self.mode = 'val' self.mode = 'val'
self.data_loader = data_loader self.data_loader = data_loader
self.call_hook('before_val_iter')
data_batch = next(data_loader) data_batch = next(data_loader)
self.call_hook('before_val_iter')
outputs = self.model.val_step(data_batch, **kwargs) outputs = self.model.val_step(data_batch, **kwargs)
if not isinstance(outputs, dict): if not isinstance(outputs, dict):
raise TypeError('model.val_step() must return a dict') raise TypeError('model.val_step() must return a dict')
......
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