Initialize model_name attribute (#19224)

2025-02-13 20:57:36 +08:00 · 2025-02-13 20:57:36 +08:00 · f4114b0319
commit f4114b0319
parent 316f6574ca
1 changed files with 1 additions and 0 deletions
--- a/ultralytics/engine/model.py
+++ b/ultralytics/engine/model.py
@ -122,6 +122,7 @@ class Model(torch.nn.Module):
        self.metrics = None  # validation/training metrics
        self.session = None  # HUB session
        self.task = task  # task type
+        self.model_name = None  # model name
        model = str(model).strip()

        # Check if Ultralytics HUB model from https://hub.ultralytics.com