noahmr/yolov5-tensorrt

convert onnx to engine failed

formance opened this issue · 1 comments

when convert yolov5s.onnx to engine, encountered workspace memory failed, how to solve it. but it worked using python script to convert the same model.
image

I have the same issue. Converts directly from the python script works. Does it make a difference?