From 828693029c0463828d9fae714c70514f9c3bf50b Mon Sep 17 00:00:00 2001 From: Frank Liu Date: Tue, 8 Jun 2021 15:29:27 -0700 Subject: [PATCH] [tensorflow] fix optOption usage document (#1001) Change-Id: Ie044839cf082d63010a5c26d3f2f8833447919c6 --- docs/tensorflow/how_to_import_tensorflow_models_in_DJL.md | 2 +- .../src/main/java/ai/djl/tensorflow/engine/TfSymbolBlock.java | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/tensorflow/how_to_import_tensorflow_models_in_DJL.md b/docs/tensorflow/how_to_import_tensorflow_models_in_DJL.md index a1e968e568d..8c373dfd4b4 100644 --- a/docs/tensorflow/how_to_import_tensorflow_models_in_DJL.md +++ b/docs/tensorflow/how_to_import_tensorflow_models_in_DJL.md @@ -121,7 +121,7 @@ Criteria criteria = .setTypes(Image.class, DetectedObjects.class) .optFilter("backbone", "mobilenet_v2") .optEngine("TensorFlow") - .optOption("Tags", new String[] {}) + .optOption("Tags", "") .optOption("SignatureDefKey", "default") .optProgress(new ProgressBar()) .build(); diff --git a/tensorflow/tensorflow-engine/src/main/java/ai/djl/tensorflow/engine/TfSymbolBlock.java b/tensorflow/tensorflow-engine/src/main/java/ai/djl/tensorflow/engine/TfSymbolBlock.java index d2c48b49133..e676f9aefc5 100644 --- a/tensorflow/tensorflow-engine/src/main/java/ai/djl/tensorflow/engine/TfSymbolBlock.java +++ b/tensorflow/tensorflow-engine/src/main/java/ai/djl/tensorflow/engine/TfSymbolBlock.java @@ -79,7 +79,7 @@ public TfSymbolBlock(SavedModelBundle bundle, String signatureDefKey) { + "not found in Saved Model Bundle." + "Available keys: " + String.join(" ", keys) - + "Please use .optOptions(\"SignatureDefKey\", \"value\") with Criteria.builder to load the model." + + "Please use .optOption(\"SignatureDefKey\", \"value\") with Criteria.builder to load the model." + "Normally the value is \"default\" for TF1.x models and \"serving_default\" for TF2.x models. " + "Refer to: https://www.tensorflow.org/guide/saved_model" + "Loading the model using next available key.");