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

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
action-recog
Mohammed Yasin 3 months ago committed by fcakyon
parent 199b1f6973
commit a43a141fc2
  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