Merge pull request #464 from AnvarNazar/patch-1

fix import
2023-09-26 08:41:22 -03:00
parent b738644f6e a9d2f13544
commit 000bcd676d
1 changed files with 1 additions and 1 deletions
--- 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