From acc94d92823caef35c8ee276d2cd7eb8b6a0afff Mon Sep 17 00:00:00 2001 From: Josh Veitch-Michaelis Date: Fri, 10 Sep 2021 13:33:06 +0000 Subject: [PATCH] add callbacks to train function in wandb sweep Fix following https://github.com/ultralytics/yolov5/pull/4688 which modified the function signature to `train` --- utils/loggers/wandb/sweep.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/utils/loggers/wandb/sweep.py b/utils/loggers/wandb/sweep.py index 2dcda508eb50..04296e64f893 100644 --- a/utils/loggers/wandb/sweep.py +++ b/utils/loggers/wandb/sweep.py @@ -9,6 +9,7 @@ from train import train, parse_opt from utils.general import increment_path from utils.torch_utils import select_device +from utils.callbacks import Callbacks def sweep(): @@ -26,7 +27,8 @@ def sweep(): device = select_device(opt.device, batch_size=opt.batch_size) # train - train(hyp_dict, opt, device) + callbacks = Callbacks() + train(hyp_dict, opt, device, callbacks) if __name__ == "__main__":