Skip to content

GitLab

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

Add torch.no_grad() decorator to the whole val workflow(#777)

parent 477f0c0a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
mmcv/runner/epoch_based_runner.py
View file @ daab369e
......@@ -54,6 +54,7 @@ class EpochBasedRunner(BaseRunner):
self.call_hook('after_train_epoch')
self._epoch += 1
@torch.no_grad()
def val(self, data_loader, **kwargs):
self.model.eval()
self.mode = 'val'
......@@ -63,8 +64,7 @@ class EpochBasedRunner(BaseRunner):
for i, data_batch in enumerate(self.data_loader):
self._inner_iter = i
self.call_hook('before_val_iter')
with torch.no_grad():
self.run_iter(data_batch, train_mode=False)
self.run_iter(data_batch, train_mode=False)
self.call_hook('after_val_iter')
self.call_hook('after_val_epoch')
......
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