如何检查张量流保存模型中的图形定义

Question

我正在使用来自 tensorflow 对象检测模型动物园的一些模型，并使用 tf-trt 进行优化。

我的问题与是否有办法以图形方式或至少以编程方式检查模型的图形定义（逐层）。我的目标是比较原始网络定义和执行后的 trt 优化。

提前感谢您的帮助！

Answer 1

您可以尝试使用 Netron 应用程序来可视化您的图表。 只需使用浏览器会话从netron github repository 上传您的模型。