mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Update sweep.py (#5402)

pull/5404/head
Ayush Chaurasia 2021-10-29 22:51:59 +05:30 committed by GitHub
parent ed887b5976
commit 620b535f85
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
utils/loggers/wandb/sweep.py
View File

@ -26,6 +26,11 @@ def sweep():
opt.epochs = hyp_dict.get("epochs") opt.epochs = hyp_dict.get("epochs")
opt.nosave = True opt.nosave = True
opt.data = hyp_dict.get("data") opt.data = hyp_dict.get("data")
opt.weights = str(opt.weights)
opt.cfg = str(opt.cfg)
opt.data = str(opt.data)
opt.hyp = str(opt.hyp)
opt.project = str(opt.project)
device = select_device(opt.device, batch_size=opt.batch_size) device = select_device(opt.device, batch_size=opt.batch_size)
# train # train