Tensorrt 7.1.3 and ONNX 1.6 issuet

AI & Data Science Deep Learning (Training & Inference) TensorRT
,
1

I use TensorRT 7.1.3 on Jetpack 4.5.1 TX2 with ONNX 1.6 but unsuccessful to convert weights from pytorch to ONNX and TensorRT. Please help me out.
The error as follows:

TensorRT: export failure \u274c 1.1s: failed to export ONNX file: runs/train/exp/weights/best.onnx

2

Hi,

This belongs in the TensorRT forum, I have moved it over for you.

3

Hi,
Request you to share the ONNX model and the script if not shared already so that we can assist you better.
Alongside you can try few things:

  1. validating your model with the below snippet

check_model.py

import sys
import onnx
filename = yourONNXmodel
model = onnx.load(filename)
onnx.checker.check_model(model).
2) Try running your model with trtexec command.

In case you are still facing issue, request you to share the trtexec “”–verbose"" log for further debugging
Thanks!

5

Hi,

If you’re facing an issue while exporting the PyTorch model to ONNX, please try following.
https://pytorch.org/tutorials/advanced/super_resolution_with_onnxruntime.html

Thank you.