mirror of
https://github.com/ultralytics/yolov5.git
synced 2025-06-03 14:49:29 +08:00
efe60b5681
* Update loggers * Config * Update val.py * cleanup * fix1 * fix2 * fix3 and reformat * format sweep.py * Logger() class * cleanup * cleanup2 * wandb package import fix * wandb package import fix2 * txt fix * fix4 * fix5 * fix6 * drop wandb into utils/loggers * fix 7 * rename loggers/wandb_logging to loggers/wandb * Update message * Update message * Update message * cleanup * Fix x axis bug * fix rank 0 issue * cleanup
34 lines
866 B
Python
34 lines
866 B
Python
import sys
|
|
from pathlib import Path
|
|
|
|
import wandb
|
|
|
|
FILE = Path(__file__).absolute()
|
|
sys.path.append(FILE.parents[2].as_posix()) # add utils/ to path
|
|
|
|
from train import train, parse_opt
|
|
from utils.general import increment_path
|
|
from utils.torch_utils import select_device
|
|
|
|
|
|
def sweep():
|
|
wandb.init()
|
|
# Get hyp dict from sweep agent
|
|
hyp_dict = vars(wandb.config).get("_items")
|
|
|
|
# Workaround: get necessary opt args
|
|
opt = parse_opt(known=True)
|
|
opt.batch_size = hyp_dict.get("batch_size")
|
|
opt.save_dir = str(increment_path(Path(opt.project) / opt.name, exist_ok=opt.exist_ok or opt.evolve))
|
|
opt.epochs = hyp_dict.get("epochs")
|
|
opt.nosave = True
|
|
opt.data = hyp_dict.get("data")
|
|
device = select_device(opt.device, batch_size=opt.batch_size)
|
|
|
|
# train
|
|
train(hyp_dict, opt, device)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sweep()
|