Fix a GPU memory leak in detect. No need to calculate gradients in inference. (WongKinYiu#900)

Co-authored-by: JKO095 <[email protected]>

Author: jpkoponen

detect.py

@@ -84,7 +84,8 @@ def detect(save_img=False):
 
         # Inference
         t1 = time_synchronized()
-        pred = model(img, augment=opt.augment)[0]
+        with torch.no_grad():   # Calculating gradients would cause a GPU memory leak
+            pred = model(img, augment=opt.augment)[0]
         t2 = time_synchronized()
 
         # Apply NMS