.half() bug fix

2020-07-05 23:24:53 -07:00 · 2020-07-05 23:24:53 -07:00 · a40f615c6f
parent 04bdbe4104
commit a40f615c6f
1 changed files with 1 additions and 1 deletions
--- a/detect.py
+++ b/detect.py
@ -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