2017-02-02 1 views
2

私はtensorflow tutorialで見ましたが、起こっている操作の数など、さまざまなアーキテクチャに関するいくつかの興味深い統計情報を提供しています。Caffeでは、アーキテクチャで実行されるオペレーションの数を計算できますか?

このモデルは、GPUでのトレーニング時間が数時間で約86%のピーク性能を実現します。詳しくは のコードをご覧ください。これは1,068,298の学習可能なパラメータで構成され、単一の イメージでの推論を計算するために約19.5Mの積和演算を で実行する必要があります。

Caffeを使用して同様のことを達成するにはどうすればよいですか?
このような興味深い統計情報を自分で計算またはコンパイルする方法はありますか?

答えて

1

(ちなみに私はtensorboardの可視化は、そのようなそのような操作の回数やものとしてだけで統計の話ではない)、この機能を要求BVLC /カフェのgithubので開いissue #2507があります。
あなたはこの機能を実装しているCaffeに貢献しています。

+0

ありがとう、私はすでにdeplot.prototxtを使用して訓練可能なパラメータの数を計算するためのスクリプトを作成しました。私はFLOPs関連の統計情報を取得する方法はわかりません! – Breeze

関連する問題