提交 a40f615c authored 作者: Glenn Jocher's avatar Glenn Jocher

.half() bug fix

上级 04bdbe41
......@@ -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
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论