models icon indicating copy to clipboard operation
models copied to clipboard

how to find the mapping relation between onnx node and pytorch layer?

Open GeneralJing opened this issue 3 years ago • 0 comments

recently, i want to analysis the time of model inference. I can see the most time consuming node op in the layer details by some tools, but the node name conv_785 is confusing. because the conv is very normal in the model structure. it's difficult for me to find the right layer in the model. is there any convinent way to find the right mapping relationship between onnx node and pytorch layer? thanks.

GeneralJing avatar Nov 11 '22 03:11 GeneralJing