Fix typo (#3729)
This commit is contained in:
1 parent
b83e1a4adc
commit
0e2d0d54d7
1 file changed
+1
-1
@@ -43,7 +43,7 @@ def run(weights='./yolov5s.pt', # weights path
|
|||||||
|
|
||||||
# Load PyTorch model
|
# Load PyTorch model
|
||||||
device = select_device(device)
|
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
|
model = attempt_load(weights, map_location=device) # load FP32 model
|
||||||
labels = model.names
|
labels = model.names
|
||||||
|
|
||||||
|
|||||||
Reference in new issue
Block a user