Open
Description
Hi, in models/yolox.py and detectron2onnx.py, we can add support to AMD GPUs by modifying ordered_providers
list to
ordered_providers = [
"ROCMExecutionProvider",
"TensorrtExecutionProvider",
"CUDAExecutionProvider"
]
Metadata
Metadata
Assignees
Labels
No labels