是否有任何工具可以在 Pytorch 和 Tensorflow 中分析 DNN 的每一层？答案

【问题标题】：Is there any tool to profile each layer of a DNN in Pytorch and Tensorflow?是否有任何工具可以在 Pytorch 和 Tensorflow 中分析 DNN 的每一层？
【发布时间】：2020-02-18 17:21:25
【问题描述】：

我想知道 DNN 每一层的执行时间。有什么工具可以分析每一层吗？我知道我可以在模型中插入一些打印件，但我不确定它是否准确。

【问题讨论】：

要求我们推荐或查找书籍、工具、软件库、教程或其他场外资源的问题对于 Stack Overflow 来说是无关紧要的，因为它们往往会吸引固执己见的答案和垃圾邮件。 What topics can I ask about here?
个人资料是什么意思？
@AndréPacheco 例如，另一个框架 Caffe 提供了“caffe time”来衡量每一层的执行时间。但是 Pytorch 和 TF 提供的工具无法提供更高层次的信息。
@LennoxWu 明白了！我知道的最接近的工具是torch summary
@AndréPacheco 谢谢。对我有用。

标签： tensorflow machine-learning pytorch

【解决方案1】：

要检查每一层的大小和参数，您可以使用torch summary。

要计算每一层的执行时间，可以使用torchprof。

我不知道任何合并这两个库的项目。也许这是你的机会（笑）

【讨论】：