glenn-jocher commited on
Commit
4f9718a
1 Parent(s): a0e1504

Pass `--device` for `--task study` (#5118)

Browse files
Files changed (1) hide show
  1. val.py +1 -1
val.py CHANGED
@@ -346,7 +346,7 @@ def main(opt):
346
  for i in x: # img-size
347
  print(f'\nRunning {f} point {i}...')
348
  r, _, t = run(opt.data, weights=w, batch_size=opt.batch_size, imgsz=i, conf_thres=opt.conf_thres,
349
- iou_thres=opt.iou_thres, save_json=opt.save_json, plots=False)
350
  y.append(r + t) # results and times
351
  np.savetxt(f, y, fmt='%10.4g') # save
352
  os.system('zip -r study.zip study_*.txt')
 
346
  for i in x: # img-size
347
  print(f'\nRunning {f} point {i}...')
348
  r, _, t = run(opt.data, weights=w, batch_size=opt.batch_size, imgsz=i, conf_thres=opt.conf_thres,
349
+ iou_thres=opt.iou_thres, device=opt.device, save_json=opt.save_json, plots=False)
350
  y.append(r + t) # results and times
351
  np.savetxt(f, y, fmt='%10.4g') # save
352
  os.system('zip -r study.zip study_*.txt')