neural-networktensorrt

How can I visualize TensorRT network graph?


Our current flow: Conversation of tf2->onnx->TensorRT (all16 and 32 and 8 bits options)

Is there an existing tool like https://github.com/lutzroeder/netron (or any other way) to see the output model of TensorRT?


Solution

  • You can now visualize tensorrt engine graphs using https://github.com/NVIDIA/TensorRT/tree/main/tools/experimental/trt-engine-explorer

    It also offers other info about the engine.