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__":