# 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 | ||||