sxy009858 2022-07-26 15:16 采纳率: 0%
浏览 9

如何得到深度学习模型的向前传播时间和模型复杂度

想要得到深度学习模型的向前传播时间和模型复杂度,请问需要如何输出,用什么函数,用什么代码?
  • 写回答

1条回答 默认 最新

  • kakaccys 2022-07-26 20:54
    关注

    首先,你用的是什么框架?当然不管是paddlepaddle还是torch亦或者是tensorflow,都有summary函数方法,通过该方法是可以知道模型大小以及复杂度的。
    至于向前传播时间,其实近似于推理时间,所以通过打印predict的时间就能知道了。

    评论

报告相同问题?

问题事件

  • 创建了问题 7月26日