UltralyticsAssistant 3 months ago
parent a5f7b7ae48
commit e02be693eb
  1. 2
      ultralytics/data/build.py
  2. 2
      ultralytics/engine/model.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, LOGGER
from ultralytics.utils import LOGGER, RANK, colorstr
from ultralytics.utils.checks import check_file

@ -800,7 +800,7 @@ class Model(nn.Module):
"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
"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"):

Loading…
Cancel
Save