Update sweep.py (#5402)

This commit is contained in:
Ayush Chaurasia authored and GitHub committed 2021-10-29 19:21:59 +02:00
1 parent ed887b5976
commit 620b535f85
1 file changed
+5
+5
View File
@@ -26,6 +26,11 @@ def sweep():
opt.epochs = hyp_dict.get("epochs")
opt.nosave = True
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)
# train