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