diff --git a/src/model/yolo_wrapper.py b/src/model/yolo_wrapper.py
index f08ef69..ea9bd42 100644
--- a/src/model/yolo_wrapper.py
+++ b/src/model/yolo_wrapper.py
@@ -100,6 +100,7 @@ class YOLOWrapper:
                 name=name,
                 device=self.device,
                 resume=resume,
+                callbacks=callbacks,
                 **kwargs,
             )