Exporting tao ssd trained models without post processing

Accelerated Computing Intelligent Video Analytics TAO Toolkit
1

Please provide the following information when requesting support.

• Hardware (Quadro GV100)
• Network Type (ssd)

I trained a model using the tao 5.3.0 using the ssd resnet18 backbone on the kitti dataset and pruned the model using the provided default spec files in the tao_getting_started-5.3/notbooks/tao_launcher_starter_kit/.
When I exported the trained model to the onnx format, I can not export it without the post processing node. Could you please help me, I want to run the exported model using simple onnxruntime (if possible).

Thank you.

2

After exporting to onnx, you can modify the onnx file by using onnx graphsurgeon tool.