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

Merge branch 'main' of github.com:jameslahm/yolov10 into main

Browse Source
This commit is contained in:
wa22 2024-06-01 19:12:41 +08:00
commit 0f87aa3173
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.