AI Model Visualization
导言
作为一个AI初学者，总是遇到以下场景：
- 客户正在基于NV开发一个AI模型，需要同步的做昇腾适配。手上只有NV下的代码。
- 往往很难将论文里的AI模型的图，和代码里的每一层以及参数对应起来。
设计期望：
- 在模型开发的过程中，能简单插入，来明确当前模块的大致信息。
- 名称，类型(卷积层，池化层)，输入/输出/参数, 执行的时间(第一次)。
- 可视化
- 格式兼容cpprinter。
- 能体现出TP，CP等并行策略的效果。
大致思路：
- 还是借助
chrome://tracing格式，来设计类似PyPrinter的工具。 - 早期可以使用
VizTracer代替。