Graphpipe向けMNISTモデルの作成

TensorFlow Deep MNIST for ML Beginners をベースとしたモデルをGraphpipeにて活用できるよう、ProtocolBufferファイルを作成します。

1.ソースコード(mnist_test_beginner.py)

ここにそのまま流用できるものがあったので、使わせていただきます。

実行し、pbファイルができることを確認します。

# python mnist_test_beginner.py
# ls -ltr ./tmp/beginner-export/
total 32
-rw-r–r–. 1 root root 31771 Dec 21 08:44 beginner-graph.pb
2.Graphpipeの実行

生成されたpbファイルを用いてDockerを起動させます。

# docker run -it –rm \
> -v “$PWD:/models/” \
> -p 9001:9001 \
> sleepsonthefloor/graphpipe-tf:cpu \
> –model=/models/beginner-graph.pb \
> –listen=0.0.0.0:9001
INFO[0000] Starting graphpipe-tf version 1.0.0.10.f235920 (built from sha f235920)
2018-12-21 08:54:45.008645: I tensorflow/core/platform/cpu_feature_guard.cc:140] Your CPU supports instructions that this TensorFlow binary was not compiled to use: SSE4.1 SSE4.2 AVX AVX2 FMA
2018-12-21 08:54:45.009619: I tensorflow/core/common_runtime/process_util.cc:63] Creating new thread pool with default inter op setting: 2. Tune using inter_op_parallelism_threads for best performance.
INFO[0000] Model hash is ‘03506033e068c4da9e8abdae16e2987915f1029f791eb803d92e0039053880e0’
INFO[0000] Using default inputs [input:0]
INFO[0000] Using default outputs [output:0]
INFO[0000] Listening on ‘0.0.0.0:9001’

正常に起動できました。

シェアする

  • このエントリーをはてなブックマークに追加

フォローする