From 9f812122a2256dfd7ccd3704ae0f97acc2983aa3 Mon Sep 17 00:00:00 2001 From: fcakyon <34196005+fcakyon@users.noreply.github.com> Date: Tue, 22 Jun 2021 14:22:01 +0300 Subject: [PATCH] fix typo --- export.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/export.py b/export.py index 8f4000cdad39..b7ff0748ba93 100644 --- a/export.py +++ b/export.py @@ -43,7 +43,7 @@ def run(weights='./yolov5s.pt', # weights path # Load PyTorch model device = select_device(device) - assert not (device.type == 'cpu' and opt.half), '--half only compatible with GPU export, i.e. use --device 0' + assert not (device.type == 'cpu' and half), '--half only compatible with GPU export, i.e. use --device 0' model = attempt_load(weights, map_location=device) # load FP32 model labels = model.names