fixed default augment to True

ultralytics/data/build.py

@@ -21,7 +21,7 @@ from ultralytics.data.loaders import (
     autocast_list,
 )
 from ultralytics.data.utils import IMG_FORMATS, PIN_MEMORY, VID_FORMATS
-from ultralytics.utils import RANK, colorstr
+from ultralytics.utils import RANK, colorstr, LOGGER
 from ultralytics.utils.checks import check_file
 
 
@@ -84,6 +84,8 @@ def seed_worker(worker_id):  # noqa
 def build_yolo_dataset(cfg, img_path, batch, data, mode="train", rect=False, stride=32, multi_modal=False):
     """Build YOLO Dataset."""
     dataset = YOLOMultiModalDataset if multi_modal else YOLODataset
+    if not cfg.augment and mode == "train":
+        LOGGER.warning(f"WARNING ⚠️ 'augment' is set to False in {mode} mode. This may reduce model performance.")
     return dataset(
         img_path=img_path,
         imgsz=cfg.imgsz,

ultralytics/engine/model.py

@@ -794,13 +794,13 @@ class Model(nn.Module):
             kwargs = self.session.train_args  # overwrite kwargs
 
         checks.check_pip_update_available()
-
         overrides = yaml_load(checks.check_yaml(kwargs["cfg"])) if kwargs.get("cfg") else self.overrides
         custom = {
             # NOTE: handle the case when 'cfg' includes 'data'.
             "data": overrides.get("data") or DEFAULT_CFG_DICT["data"] or TASK2DATA[self.task],
             "model": self.overrides["model"],
             "task": self.task,
+            "augment": kwargs.get("augment", True), # default to True for training
         }  # method defaults
         args = {**overrides, **custom, **kwargs, "mode": "train"}  # highest priority args on the right
         if args.get("resume"):