Apply no_grad in the inference phase

nmhkahn · Nov 6, 2018 · 5758408 · 5758408
1 parent f5a8319
commit 5758408
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/carnpp/solver.py b/carnpp/solver.py
@@ -100,7 +100,8 @@ def evaluate(self, test_data, scale=2):
         for step, inputs in enumerate(test_loader):
             HR = inputs[0].to(self.device)
             LR = inputs[1].to(self.device)
-            SR = self.net(LR, scale).detach()
+            with torch.no_grad():
+                SR = self.net(LR, scale).detach()
 
             HR = HR.cpu().clamp(0, 1).squeeze(0).permute(1, 2, 0).numpy()
             SR = SR.cpu().clamp(0, 1).squeeze(0).permute(1, 2, 0).numpy()