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

Add mixin

Browse Source
This commit is contained in:
Niels 2024-06-01 09:35:37 +02:00
parent 22aa8412a7
commit 4b118ae016
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ultralytics/engine/model.py
View File

@ -13,8 +13,10 @@ from ultralytics.hub.utils import HUB_WEB_ROOT
from ultralytics.nn.tasks import attempt_load_one_weight, guess_model_task, nn, yaml_model_load
from ultralytics.utils import ASSETS, DEFAULT_CFG_DICT, LOGGER, RANK, SETTINGS, callbacks, checks, emojis, yaml_load
from huggingface_hub import PyTorchModelHubMixin
class Model(nn.Module):
class Model(nn.Module, PyTorchModelHubMixin):
"""
A base class for implementing YOLO models, unifying APIs across different model types.