Skip to content

Commit

Permalink
Pass --device for --task study (ultralytics#5118)
Browse files Browse the repository at this point in the history
  • Loading branch information
glenn-jocher committed Oct 10, 2021
1 parent b2dc5a2 commit 7f4b615
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion val.py
Original file line number Diff line number Diff line change
Expand Up @@ -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')
Expand Down

0 comments on commit 7f4b615

Please sign in to comment.