xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 21:44:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix resume training

Browse Source
This commit is contained in:
wa22 2024-05-24 15:27:28 +00:00
parent a02c5bac11
commit 4197c99fbf
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ultralytics/cfg/__init__.py
View File

@ -536,7 +536,8 @@ def entrypoint(debug=""):
model = "yolov8n.pt" model = "yolov8n.pt"
LOGGER.warning(f"WARNING ⚠️ 'model' argument is missing. Using default 'model={model}'.") LOGGER.warning(f"WARNING ⚠️ 'model' argument is missing. Using default 'model={model}'.")
overrides["model"] = model overrides["model"] = model
stem = Path(model).stem.lower() # stem = Path(model).stem.lower()
stem = model.lower()
if "rtdetr" in stem: # guess architecture if "rtdetr" in stem: # guess architecture
from ultralytics import RTDETR from ultralytics import RTDETR