`ultralytics 8.2.91` fix `v10DetectLoss` module rename for YOLOv10 (#16148)

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
security-test v8.2.91
Mohammed Yasin 5 months ago committed by GitHub
parent 090db8ac3f
commit e97782943b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 2
      ultralytics/__init__.py
  2. 1
      ultralytics/nn/tasks.py

@ -1,6 +1,6 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license
__version__ = "8.2.90"
__version__ = "8.2.91"
import os

@ -776,6 +776,7 @@ def torch_safe_load(weight):
attributes={
"ultralytics.nn.modules.block.Silence": "torch.nn.Identity", # YOLOv9e
"ultralytics.nn.tasks.YOLOv10DetectionModel": "ultralytics.nn.tasks.DetectionModel", # YOLOv10
"ultralytics.utils.loss.v10DetectLoss": "ultralytics.utils.loss.E2EDetectLoss", # YOLOv10
},
):
ckpt = torch.load(file, map_location="cpu")

Loading…
Cancel
Save