From 3926bca351f24c060f837a0d0e1994e28bace1af Mon Sep 17 00:00:00 2001 From: MichaelAnderson-AI Date: Tue, 22 Jun 2021 14:33:38 +0300 Subject: [PATCH] Fix typo (#3729) --- export.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/export.py b/export.py index 8f4000c..b7ff074 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