From 85a3eafcb583ced4d75fa612e3af9829b7621543 Mon Sep 17 00:00:00 2001
From: hzwer <598460606@163.com>
Date: Wed, 30 Dec 2020 23:42:55 +0800
Subject: [PATCH] Update inference_img.py

---
 inference_img.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/inference_img.py b/inference_img.py
index 3d44e2d..1e6ea25 100644
--- a/inference_img.py
+++ b/inference_img.py
@@ -8,8 +8,8 @@ import warnings
 warnings.filterwarnings("ignore")
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+torch.set_grad_enabled(False)
 if torch.cuda.is_available():
-    torch.set_grad_enabled(False)
     torch.backends.cudnn.enabled = True
     torch.backends.cudnn.benchmark = True