From 4197c99fbfc2da7a23ff5282519dbee40ff97207 Mon Sep 17 00:00:00 2001 From: wa22 Date: Fri, 24 May 2024 15:27:28 +0000 Subject: [PATCH] fix resume training --- ultralytics/cfg/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ultralytics/cfg/__init__.py b/ultralytics/cfg/__init__.py index 175272ff..0e081d1c 100644 --- a/ultralytics/cfg/__init__.py +++ b/ultralytics/cfg/__init__.py @@ -536,7 +536,8 @@ def entrypoint(debug=""): model = "yolov8n.pt" LOGGER.warning(f"WARNING ⚠️ 'model' argument is missing. Using default 'model={model}'.") overrides["model"] = model - stem = Path(model).stem.lower() + # stem = Path(model).stem.lower() + stem = model.lower() if "rtdetr" in stem: # guess architecture from ultralytics import RTDETR