2017-06-18 44 views
1

我已經使用SavedModel(Inception_resnet_v2)導出TensorFlow模型文件,並使用TensorFlow服務加載files.I有直接取代官方minst saved_model.pb用我自己的Inception_resnet_v2 saved_model.pb文件。但我有一個錯誤。不能在tensorflow服務加載PB文件

[email protected]:~/serving$ bazel-bin/tensorflow_serving/model_servers/tensorflow_model_server --port=9000 --model_name=mnist --model_base_path=/home/deep/serving/tmp/mnist_model 2017-06-18 10:39:41.963490: I tensorflow_serving/model_servers/main.cc:146] Building single TensorFlow model file config: model_name: mnist model_base_path: home/deep/serving/tmp/mnist_model model_version_policy: 0 2017-06-18 10:39:41.963752: I tensorflow_serving/model_servers/server_core.cc:375] Adding/updating models. 2017-06-18 10:39:41.963762: I tensorflow_serving/model_servers/server_core.cc:421] (Re-)adding model: mnist 2017-06-18 10:39:42.065556: I tensorflow_serving/core/basic_manager.cc:698] Successfully reserved resources to load servable {name: mnist version: 1} 2017-06-18 10:39:42.065610: I tensorflow_serving/core/loader_harness.cc:66] Approving load for servable version {name: mnist version: 1} 2017-06-18 10:39:42.065648: I tensorflow_serving/core/loader_harness.cc:74] Loading servable version {name: mnist version: 1} 2017-06-18 10:39:42.065896: I external/org_tensorflow/tensorflow/contrib/session_bundle/bundle_shim.cc:360] Attempting to load native SavedModelBundle in bundle-shim from: /home/deep/serving/tmp/mnist_model/1 2017-06-18 10:39:42.066130: I external/org_tensorflow/tensorflow/cc/saved_model/loader.cc:226] Loading SavedModel from: /home/deep/serving/tmp/mnist_model/1 2017-06-18 10:39:42.080775: I external/org_tensorflow/tensorflow/cc/saved_model/loader.cc:274] Loading SavedModel: fail. Took 14816 microseconds. 2017-06-18 10:39:42.080822: E tensorflow_serving/util/retrier.cc:38] Loading servable: {name: mnist version: 1} failed: Not found: Could not find meta graph def matching supplied tags. 我該怎麼辦?謝謝!

回答

2

我聊到服務工程師,這裏是他們的一些想法在這:

看起來他們需要指定無論是在保存的模型,或者 命令行的標籤。 (注意日誌行:失敗:找不到:找不到 元高清圖形匹配提供的標籤。)

它看起來像SavedModel加載器是無法找到對應於他們所提供的標籤的圖形 。下面是一些 文檔: https://github.com/tensorflow/tensorflow/tree/master/tensorflow/python/saved_model#tags

啊,補充:他們可以使用SavedModel CLI檢查模型和 看看標籤集可用。以下是該文檔: https://www.tensorflow.org/versions/master/programmers_guide/saved_model_cli

它們可以運行

saved_model_cli show --dir <SavedModelDir> 

檢查什麼標籤集是在SavedModel如果他們有點子 安裝tensorflow。