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

2021-02-18 22:02:11 +08:00 · 2021-02-18 22:02:11 +08:00 · f75a88c297
parent 3d229dbaf2
commit f75a88c297
1 changed files with 1 additions and 0 deletions
--- a/mmcv/runner/iter_based_runner.py
+++ b/mmcv/runner/iter_based_runner.py
@ -67,6 +67,7 @@ class IterBasedRunner(BaseRunner):
        self._inner_iter += 1
        self._iter += 1

+    @torch.no_grad()
    def val(self, data_loader, **kwargs):
        self.model.eval()
        self.mode = 'val'