.half() bug fix

This commit is contained in:
Glenn Jocher committed 2020-07-05 23:24:53 -07:00
1 parent 04bdbe4104
commit a40f615c6f
1 file changed
+1 -1
+1 -1
View File
@@ -24,7 +24,7 @@ def detect(save_img=False):
model = torch.load(weights, map_location=device)['model'].float().eval() # load FP32 model
imgsz = check_img_size(imgsz, s=model.stride.max()) # check img_size
if half:
model.float() # to FP16
model.half() # to FP16
# Second-stage classifier
classify = False