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