【问题标题】:Is it possible in Caffe to calculate number of Operations taking place in an architecture? [duplicate]在 Caffe 中是否可以计算架构中发生的操作数? [复制]
【发布时间】:2017-06-19 04:37:12
【问题描述】:

我在tensorflow tutorials 看到过,它们提供了一些关于不同架构的有趣统计数据,例如发生的操作数量等。

该模型在一个 在 GPU 上进行几个小时的训练。请看下面和代码 细节。它由 1,068,298 个可学习参数组成,需要 大约 1950 万次乘加运算来计算单次推理 图片。

如何使用 Caffe 实现类似的功能?
如何自己计算或编译这些有趣的统计数据?
(顺便说一句,我不是在谈论张量板可视化,只是统计数据,例如操作数量之类的东西)

【问题讨论】:

    标签: caffe


    【解决方案1】:

    在 BVLC/caffe github 中有一个开放的 issue #2507 请求此功能。
    非常欢迎您为实现此功能的 caffe 做出贡献!

    【讨论】:

    • 谢谢,我已经创建了一个脚本来使用 deplot.prototxt 计算可训练参数的数量。但我不知道如何获得与 FLOPs 相关的统计数据!
    猜你喜欢
    • 2017-11-14
    • 1970-01-01
    • 2013-02-20
    • 2018-01-19
    • 1970-01-01
    • 2011-09-01
    • 2018-11-30
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多