Brother_Mumu

Hello, I found this is a problem caused by the multi-gpu defined in line 69 of lib/core.py ` # multi-gpu` ` network = torch.nn.DataParallel(network)` I comment this line to switch...

@kumardesappan Thank you for your reply! To update my progress, I tried the ARM only mode by passing "-d" as an argument and it was successful. Then I move to...

@kumardesappan I provide four files for reproduce the above mentioned issue: - "onnxrt_ep_mymodel.py" : the script for compling my onnx models "psp-r50-512x512onnx.onnx" - "utils.py": add information such as mean value...