diff --git a/mmcv/runner/epoch_based_runner.py b/mmcv/runner/epoch_based_runner.py index 1ec616ce13..1e1de295ed 100644 --- a/mmcv/runner/epoch_based_runner.py +++ b/mmcv/runner/epoch_based_runner.py @@ -47,7 +47,7 @@ def train(self, data_loader, **kwargs): for i, data_batch in enumerate(self.data_loader): self._inner_iter = i self.call_hook('before_train_iter') - self.run_iter(data_batch, train_mode=True) + self.run_iter(data_batch, train_mode=True, **kwargs) self.call_hook('after_train_iter') self._iter += 1