mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

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

pull/842/head
Zhiyuan Chen 2021-02-18 22:02:11 +08:00 committed by GitHub
parent 3d229dbaf2
commit f75a88c297
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mmcv/runner/iter_based_runner.py
View File

@ -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'