Tensorflowでの8ビット量子化エラー

Question

TensorFlowで利用可能な新しい8-bit quantization featureを試しています。私はブログの投稿（googlenetの量子化）の問題をせずに与えられた例を実行することができ、それは私のために正常に動作します！

ここでは、よりシンプルなネットワークにも同じものを適用したいと考えています。 CIFAR-10（Caffeの訓練を受けています）のための事前訓練を受けたネットワークを使用し、そのパラメータを抽出し、対応するグラフをテンソルフローで作成し、この訓練された重みで初期化し、最後にGraphDefオブジェクトとして保存しました。完全な手順についてはIPython Notebookを参照してください。ピート・ウォーデンのブログで述べたように

は、今私はtensorflowスクリプトで8ビットの量子化を適用：

bazel-bin/tensorflow/contrib/quantization/tools/quantize_graph --input=cifar.pb --output=qcifar.pb --mode=eightbit --bitdepth=8 --output_node_names="ArgMax" 

は、今私は、この量子化されたネットワーク上の分類を実行したかったです。だから私は新しいqcifar.pbをテンソルフローのセッションにロードし、画像を渡しました（元のバージョンに渡したのと同じ方法で）。完全なコードは、このIPython Notebook.

で見つけることができます。しかし、あなたが最後に見ることができるように、私は次のようなエラーになっています：

NotFoundError: Op type not registered 'QuantizeV2'

は、誰もが私がここで行方不明です何をお勧めすることはできますか？

Answer 1

量子化されたopsとカーネルがcontribにあるため、Pythonスクリプトに明示的にロードする必要があります。その例がthe quantize_graph.py script itselfにあります：

from tensorflow.contrib.quantization import load_quantized_ops_so from tensorflow.contrib.quantization.kernels import load_quantized_kernels_so

これは、我々が言及する文書を更新する必要がありますものです！