How to visualize TAO Toolkit trained model architecture

Accelerated Computing Intelligent Video Analytics TAO Toolkit
1

• Hardware: RTX 3060
• Network Type: EmotionNet
• TAO Version: 4.0.1
• Training spec file: emotionnet_tlt_pretrain.zip (1.1 KB)

For my drowsiness detection project, I have retrained Emotion Net on TAO Toolkit to classify only two emotions, being “awake” or “drowsy”. How can I visualize the original (Emotion Net) and the new neural network architectures? Thanks!

3

There is no update from you for a period, assuming this is not an issue anymore. Hence we are closing this topic. If need further support, please open a new one. Thanks

You can export to .etlt file and then decode it into .onnx file via the guide from https://github.com/NVIDIA-AI-IOT/tao_toolkit_recipes/blob/main/tao_forum_faq/FAQ.md#tlt-or-etlt.
Then use Netron to visualize the onnx file.

4

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.