2017-02-02 4 views
2

Я видел в tensorflow tutorial s, они предоставляют некоторую интересную статистику о разных архитектурах, таких как количество выполняемых операций и т. Д.Возможно ли в Caffe рассчитать количество операций, происходящих в архитектуре?

Эта модель достигает максимальной производительности около 86% точности в пределах нескольких часов обучения на графическом процессоре. См. Ниже и код для . Он состоит из 1 068 298 обучаемых параметров и требует около 19,5 млн операций умножения для вычисления вывода на одном изображении .

Как я могу достичь чего-то подобного с помощью Caffe?
Как можно вычислить или скомпилировать такую ​​интересную статистику самостоятельно?
(кстати, я не говорю о tensorboard визуализаций, просто статистика, такие как количество операций и тому подобное)

ответ

1

Существует в BVLC/Caffe GitHub запрашивающей эту функцию открытой issue #2507.
Вы более чем можете внести свой вклад в создание этой функции!

+0

Спасибо, я уже создал скрипт для вычисления числа обучаемых параметров с помощью deplot.prototxt. но я не знаю, как получить статистику, связанную с FLOPs! – Breeze