Hi. Probably you’re getting “failed to export onnx file” because you need to run pip3 install onnx, if it doesn’t solve the problem, please share how you are converting your pt model. About the assertation error, I had the same problem on Jetson Nano, but I was doing inference through torch hub:
model = torch.hub.load(‘yolov5’, ‘custom’, path=‘path to your .engine model’, source=‘local’)
# passing img_size as an argument solved the problem for me
result = model(img, img_size)
You can find more about torch hub here