From 4b60b24f06dbab541716416e5dfea16330edad75 Mon Sep 17 00:00:00 2001 From: Devendra Parihar <54232149+Devparihar5@users.noreply.github.com> Date: Fri, 6 Oct 2023 06:55:39 +0000 Subject: [PATCH] fix numerical stability issues in half-precision --- detect.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/detect.py b/detect.py index 5e0c4416a4..bd35c4e2af 100644 --- a/detect.py +++ b/detect.py @@ -28,7 +28,7 @@ def detect(save_img=False): # Initialize set_logging() device = select_device(opt.device) - half = device.type != 'cpu' # half precision only supported on CUDA + half = False # fix numerical stability issues in half-precision # Load model model = attempt_load(weights, map_location=device) # load FP32 model