From acc94d92823caef35c8ee276d2cd7eb8b6a0afff Mon Sep 17 00:00:00 2001 From: Josh Veitch-Michaelis Date: Fri, 10 Sep 2021 13:33:06 +0000 Subject: [PATCH 1/2] 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__": From a69f50dc68f3695cf1db13bc562043250206bb6a Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 10 Sep 2021 16:18:54 +0200 Subject: [PATCH 2/2] Cleanup --- utils/loggers/wandb/sweep.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/utils/loggers/wandb/sweep.py b/utils/loggers/wandb/sweep.py index 04296e64f893..0ca704dd28dd 100644 --- a/utils/loggers/wandb/sweep.py +++ b/utils/loggers/wandb/sweep.py @@ -27,8 +27,7 @@ def sweep(): device = select_device(opt.device, batch_size=opt.batch_size) # train - callbacks = Callbacks() - train(hyp_dict, opt, device, callbacks) + train(hyp_dict, opt, device, callbacks=Callbacks()) if __name__ == "__main__":