Browse Source

Update sweep.py (#5402)

modifyDataloader
Ayush Chaurasia GitHub 3 years ago
parent
commit
620b535f85
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 0 deletions
  1. +5
    -0
      utils/loggers/wandb/sweep.py

+ 5
- 0
utils/loggers/wandb/sweep.py 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

Loading…
Cancel
Save