UltralyticsAssistant 3 months ago
parent 709b3fd3aa
commit 51842d7b01
  1. 2
      ultralytics/nn/autobackend.py

@ -178,7 +178,7 @@ class AutoBackend(nn.Module):
providers = ["CUDAExecutionProvider", "CPUExecutionProvider"] if cuda else ["CPUExecutionProvider"] providers = ["CUDAExecutionProvider", "CPUExecutionProvider"] if cuda else ["CPUExecutionProvider"]
if mct: if mct:
import mct_quantizers as mctq import mct_quantizers as mctq
from sony_custom_layers.pytorch.object_detection import nms_ort
session = onnxruntime.InferenceSession( session = onnxruntime.InferenceSession(
w, mctq.get_ort_session_options(), providers=["CPUExecutionProvider"] w, mctq.get_ort_session_options(), providers=["CPUExecutionProvider"]
) )

Loading…
Cancel
Save