diff --git a/ultralytics/__init__.py b/ultralytics/__init__.py index b45d97ce59..5db8585676 100644 --- a/ultralytics/__init__.py +++ b/ultralytics/__init__.py @@ -1,6 +1,6 @@ # Ultralytics YOLO 🚀, AGPL-3.0 license -__version__ = "8.2.90" +__version__ = "8.2.91" import os diff --git a/ultralytics/nn/tasks.py b/ultralytics/nn/tasks.py index 07a040d90d..46974620a6 100644 --- a/ultralytics/nn/tasks.py +++ b/ultralytics/nn/tasks.py @@ -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")