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

Question

这个问题在这里已经有了答案:





how to calculate a net's FLOPs in CNN [closed]

(4 个回答)


4年前关闭。




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

This model achieves a peak performance of about 86% accuracy within a few hours of training time on a GPU. Please see below and the code for details. It consists of 1,068,298 learnable parameters and requires about 19.5M multiply-add operations to compute inference on a single image.

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

Answer 1

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