with no_grad at evaluate

Hiroshiba · Apr 25, 2020 · 442058b · 442058b
1 parent fbaa3ef
commit 442058b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pytorch_trainer/training/extensions/evaluator.py b/pytorch_trainer/training/extensions/evaluator.py
@@ -173,7 +173,8 @@ def __call__(self, trainer=None):
                                    target.named_children())
 
         with reporter:
-            result = self.evaluate()
+            with torch.no_grad():
+                result = self.evaluate()
 
         reporter_module.report(result)
         return result