glenn-jocher commited on
Commit
b0ade48
·
unverified ·
1 Parent(s): cce7e78

Fix missing `opt.device` on `--task study` (#5031)

Browse files
Files changed (1) hide show
  1. val.py +1 -1
val.py CHANGED
@@ -335,7 +335,7 @@ def main(opt):
335
  elif opt.task == 'speed': # speed benchmarks
336
  for w in opt.weights if isinstance(opt.weights, list) else [opt.weights]:
337
  run(opt.data, weights=w, batch_size=opt.batch_size, imgsz=opt.imgsz, conf_thres=.25, iou_thres=.45,
338
- save_json=False, plots=False)
339
 
340
  elif opt.task == 'study': # run over a range of settings and save/plot
341
  # python val.py --task study --data coco.yaml --iou 0.7 --weights yolov5s.pt yolov5m.pt yolov5l.pt yolov5x.pt
 
335
  elif opt.task == 'speed': # speed benchmarks
336
  for w in opt.weights if isinstance(opt.weights, list) else [opt.weights]:
337
  run(opt.data, weights=w, batch_size=opt.batch_size, imgsz=opt.imgsz, conf_thres=.25, iou_thres=.45,
338
+ device=opt.device, save_json=False, plots=False)
339
 
340
  elif opt.task == 'study': # run over a range of settings and save/plot
341
  # python val.py --task study --data coco.yaml --iou 0.7 --weights yolov5s.pt yolov5m.pt yolov5l.pt yolov5x.pt