From 802187bd928afd8cdbda101a9f61e614517f2c84 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 2 Oct 2021 17:55:40 -0700 Subject: [PATCH] Fix missing `opt.device` on `--task study` --- val.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/val.py b/val.py index 4f0b49ae2ca7..6b709ce9ee43 100644 --- a/val.py +++ b/val.py @@ -335,7 +335,7 @@ def main(opt): elif opt.task == 'speed': # speed benchmarks for w in opt.weights if isinstance(opt.weights, list) else [opt.weights]: run(opt.data, weights=w, batch_size=opt.batch_size, imgsz=opt.imgsz, conf_thres=.25, iou_thres=.45, - save_json=False, plots=False) + device=opt.device, save_json=False, plots=False) elif opt.task == 'study': # run over a range of settings and save/plot # python val.py --task study --data coco.yaml --iou 0.7 --weights yolov5s.pt yolov5m.pt yolov5l.pt yolov5x.pt