check each layer's data

Robotics & Edge Computing Jetson & Embedded Systems Jetson Nano
1

Hi,

We want to check each layer’s weights and other data of our network models in TensorRT, does TensorRT have some methods to view the data? Thanks

2

Hi,

We don’t have a visualization tool for TensorRT.
But you can output the tensor data directly.

Thanks.

3

I recommend converting the network to a format recognized by DIGITS; DIGITS has a very good visualization function!

4

Hi，
1）Can you explain more specific about outputting the tensor data directly or show an example？

2）I will check the DIGITS

Thanks

5

Hi,

DIGITS is a good tool for visualizing but it doesn’t support TensoRT.

If you want to verify the training process, DIGITS is good.
But for the behavior of TensorRT engine, you will need to output the intermediate data directly.

For example, checking the layer called “prob”.

...
network->markOutput("prob");
...

const int outputIndex = engine->getBindingIndex("prob");
...
context->enqueue(...)
buffers.copyOutputToHostAsync(stream)

-> check buffers[outputIndex]

Thanks.

6

Thanks for your reply