|
|
@ -178,6 +178,7 @@ class AutoBackend(nn.Module): |
|
|
|
|
|
|
|
|
|
|
|
providers = ["CUDAExecutionProvider", "CPUExecutionProvider"] if cuda else ["CPUExecutionProvider"] |
|
|
|
providers = ["CUDAExecutionProvider", "CPUExecutionProvider"] if cuda else ["CPUExecutionProvider"] |
|
|
|
if mct: |
|
|
|
if mct: |
|
|
|
|
|
|
|
check_requirements(["mct-nightly","sony-custom-layers[torch]"]) |
|
|
|
LOGGER.info(f"Loading {w} for ONNX MCT quantization inference...") |
|
|
|
LOGGER.info(f"Loading {w} for ONNX MCT quantization inference...") |
|
|
|
import mct_quantizers as mctq |
|
|
|
import mct_quantizers as mctq |
|
|
|
from sony_custom_layers.pytorch.object_detection import nms_ort # noqa |
|
|
|
from sony_custom_layers.pytorch.object_detection import nms_ort # noqa |
|
|
|