"docs/vscode:/vscode.git/clone" did not exist on "2831f11abcb9ec7b951b6bbbcb7a85b79ee2fd79"
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
......@@ -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'
......
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