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.");