Update inference.py

remove torch.no_grad()
Media-Smart · Oct 27, 2020 · 2bf7bfd · 2bf7bfd
1 parent 345db0f
commit 2bf7bfd
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/flexinfer/inference/inference.py b/flexinfer/inference/inference.py
@@ -22,9 +22,8 @@ def __call__(self, imgs):
         Returns:
             outp (torch.float32)
         """
-        with torch.no_grad():
-            imgs = imgs.cuda()
-            outp = self.model(imgs)
+        imgs = imgs.cuda()
+        outp = self.model(imgs)
 
         return outp