diff --git a/utils/export_yoloV8.py b/utils/export_yoloV8.py index 232b722..60af34a 100644 --- a/utils/export_yoloV8.py +++ b/utils/export_yoloV8.py @@ -7,7 +7,7 @@ import torch import torch.nn as nn from copy import deepcopy from ultralytics import YOLO -from ultralytics.yolo.utils.torch_utils import select_device +from ultralytics.utils.torch_utils import select_device from ultralytics.nn.modules import C2f, Detect, RTDETRDecoder