Quantcast
Channel: TAO Toolkit - NVIDIA Developer Forums
Viewing all articles
Browse latest Browse all 497

Exporting tao ssd trained models without post processing

$
0
0

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 posts - 2 participants

Read full topic


Viewing all articles
Browse latest Browse all 497

Trending Articles