Skip to content

GitLab

Unverified Commit 47d29795 authored by zhouzaida's avatar zhouzaida Committed by GitHub
Browse files

[Fix] Fix train error EpochBasedRunner (#942)

parent 8d7e0a6d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
mmcv/runner/epoch_based_runner.py
View file @ 47d29795
...@@ -47,7 +47,7 @@ class EpochBasedRunner(BaseRunner): ...@@ -47,7 +47,7 @@ class EpochBasedRunner(BaseRunner):
for i, data_batch in enumerate(self.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')
self.run_iter(data_batch, train_mode=True) self.run_iter(data_batch, train_mode=True, **kwargs)
self.call_hook('after_train_iter') self.call_hook('after_train_iter')
self._iter += 1 self._iter += 1
......
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