在Caffe中是否可以计算架构中发生的操作数量?

时间:2017-02-02 07:39:23

标签: caffe

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

  

该型号在a内实现了约86%精度的峰值性能   GPU上几小时的培训时间。请参阅下面的代码   细节。它包含1,068,298个可学习的参数和要求   大约19.5M乘法加法运算来计算单个推理   图像。

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

1 个答案:

答案 0 :(得分:1)

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