Is there a way to view the layers and sizes thereof after parsing onnx?

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

In human readable format?

Thanks in advance,

Roos

#2

Hi,

This related thread has some details on how to dump network info into a readable JSON format: https://devtalk.nvidia.com/default/topic/1064669/tensorrt/troubleshooting-suggestions-for-onnx-v-tensorrt-discrepancies/post/5392296/#5392296

The code is also packaged up here for convenience: https://github.com/rmccorm4/tensorrt-utils/blob/master/network/dump_network.py

#3

Thanks a lot. You put me on the right track.(I’m using the c++ interface).

Greetings,

Roos