Pass callbacks to validator (#7320)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>

ultralytics/models/yolo/classify/train.py

@@ -110,7 +110,7 @@ class ClassificationTrainer(BaseTrainer):
     def get_validator(self):
         """Returns an instance of ClassificationValidator for validation."""
         self.loss_names = ['loss']
-        return yolo.classify.ClassificationValidator(self.test_loader, self.save_dir)
+        return yolo.classify.ClassificationValidator(self.test_loader, self.save_dir, _callbacks=self.callbacks)
 
     def label_loss_items(self, loss_items=None, prefix='train'):
         """

ultralytics/models/yolo/detect/train.py

@@ -89,7 +89,10 @@ class DetectionTrainer(BaseTrainer):
     def get_validator(self):
         """Returns a DetectionValidator for YOLO model validation."""
         self.loss_names = 'box_loss', 'cls_loss', 'dfl_loss'
-        return yolo.detect.DetectionValidator(self.test_loader, save_dir=self.save_dir, args=copy(self.args))
+        return yolo.detect.DetectionValidator(self.test_loader,
+                                              save_dir=self.save_dir,
+                                              args=copy(self.args),
+                                              _callbacks=self.callbacks)
 
     def label_loss_items(self, loss_items=None, prefix='train'):
         """

ultralytics/models/yolo/pose/train.py

@@ -49,7 +49,10 @@ class PoseTrainer(yolo.detect.DetectionTrainer):
     def get_validator(self):
         """Returns an instance of the PoseValidator class for validation."""
         self.loss_names = 'box_loss', 'pose_loss', 'kobj_loss', 'cls_loss', 'dfl_loss'
-        return yolo.pose.PoseValidator(self.test_loader, save_dir=self.save_dir, args=copy(self.args))
+        return yolo.pose.PoseValidator(self.test_loader,
+                                       save_dir=self.save_dir,
+                                       args=copy(self.args),
+                                       _callbacks=self.callbacks)
 
     def plot_training_samples(self, batch, ni):
         """Plot a batch of training samples with annotated class labels, bounding boxes, and keypoints."""

ultralytics/models/yolo/segment/train.py

@@ -40,7 +40,10 @@ class SegmentationTrainer(yolo.detect.DetectionTrainer):
     def get_validator(self):
         """Return an instance of SegmentationValidator for validation of YOLO model."""
         self.loss_names = 'box_loss', 'seg_loss', 'cls_loss', 'dfl_loss'
-        return yolo.segment.SegmentationValidator(self.test_loader, save_dir=self.save_dir, args=copy(self.args))
+        return yolo.segment.SegmentationValidator(self.test_loader,
+                                                  save_dir=self.save_dir,
+                                                  args=copy(self.args),
+                                                  _callbacks=self.callbacks)
 
     def plot_training_samples(self, batch, ni):
         """Creates a plot of training sample images with labels and box coordinates."""