HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Initialize model_name attribute (#19224)

Browse source
This commit is contained in:
Love and hope 2025-02-13 20:57:36 +08:00 • committed by GitHub
parent 316f6574ca
commit f4114b0319
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
ultralytics/engine/model.py
View file

@ -122,6 +122,7 @@ class Model(torch.nn.Module):
self.metrics = None # validation/training metrics self.metrics = None # validation/training metrics
self.session = None # HUB session self.session = None # HUB session
self.task = task # task type self.task = task # task type
self.model_name = None # model name
model = str(model).strip() model = str(model).strip()
# Check if Ultralytics HUB model from https://hub.ultralytics.com # Check if Ultralytics HUB model from https://hub.ultralytics.com