Skip to content

GitLab

Unverified Commit f75a88c2 authored by Zhiyuan Chen's avatar Zhiyuan Chen Committed by GitHub
Browse files

fix no torch.no_grad decorator for val in iter_based_runner (#835)

parent 3d229dba
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
mmcv/runner/iter_based_runner.py
View file @ f75a88c2
...@@ -67,6 +67,7 @@ class IterBasedRunner(BaseRunner): ...@@ -67,6 +67,7 @@ class IterBasedRunner(BaseRunner):
self._inner_iter += 1 self._inner_iter += 1
self._iter += 1 self._iter += 1
@torch.no_grad()
def val(self, data_loader, **kwargs): def val(self, data_loader, **kwargs):
self.model.eval() self.model.eval()
self.mode = 'val' self.mode = 'val'
......
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