AI Model Visualization

导言

作为一个AI初学者，总是遇到以下场景：

1. 客户正在基于NV开发一个AI模型，需要同步的做昇腾适配。手上只有NV下的代码。
2. 往往很难将论文里的AI模型的图，和代码里的每一层以及参数对应起来。

设计期望：

1. 在模型开发的过程中，能简单插入，来明确当前模块的大致信息。
   1. 名称，类型(卷积层，池化层)，输入/输出/参数, 执行的时间(第一次)。
2. 可视化
3. 格式兼容cpprinter。
4. 能体现出TP，CP等并行策略的效果。

大致思路：

• 还是借助chrome://tracing格式，来设计类似PyPrinter的工具。
• 早期可以使用VizTracer代替。

VizTracer

1.

参考文献