Custom trained model on Jetson Nano

[AastaLLL]

Hi,

Your ONNX model uses a dynamic shape, but the corresponding information (profile) doesn’t be set.
In general, you don’t need the dynamic shape support if the image size won’t change at runtime.
(it is 1024x1024x3 in your use case)

To avoid the error, please specified the batch size in the ONNX model with the below steps:

Error when convert onnxt to tensorRT with batch size more than 1

Hi, Would you mind the check this script? import onnx_graphsurgeon as gs import onnx batch = 2 graph = gs.import_onnx(onnx.load("arcface_mobile.onnx")) for inp in graph.inputs: inp.shape[0] = batch for out in graph.outputs: out.shape[0] = batch onnx.save(gs.export_onnx(graph), "arcface_mobile_bs.onnx") Thanks.

Thanks.