From 7f4b615eea91a48b4c2ec0c0ec580d96ada80fc7 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 10 Oct 2021 15:07:26 -0700 Subject: [PATCH] Pass `--device` for `--task study` (#5118) --- val.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/val.py b/val.py index 6b709ce9ee43..7ce9a47ce36e 100644 --- a/val.py +++ b/val.py @@ -346,7 +346,7 @@ def main(opt): for i in x: # img-size print(f'\nRunning {f} point {i}...') r, _, t = run(opt.data, weights=w, batch_size=opt.batch_size, imgsz=i, conf_thres=opt.conf_thres, - iou_thres=opt.iou_thres, save_json=opt.save_json, plots=False) + iou_thres=opt.iou_thres, device=opt.device, save_json=opt.save_json, plots=False) y.append(r + t) # results and times np.savetxt(f, y, fmt='%10.4g') # save os.system('zip -r study.zip study_*.txt')