From f4114b031915bafdb94d95f310ce6f4ccca259bd Mon Sep 17 00:00:00 2001 From: Love and hope <33247520+LoveAndHope-dev@users.noreply.github.com> Date: Thu, 13 Feb 2025 20:57:36 +0800 Subject: [PATCH] Initialize `model_name` attribute (#19224) --- ultralytics/engine/model.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ultralytics/engine/model.py b/ultralytics/engine/model.py index 940b9a6f..8effe39c 100644 --- 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