Skip to content

GitLab

You need to sign in or sign up before continuing.
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