From 0eff4f359c6e65fbdb6c4524379432be09960b68 Mon Sep 17 00:00:00 2001
From: liuqi <liuqi10@xiaomi.com>
Date: Tue, 30 Jan 2018 19:39:00 +0800
Subject: [PATCH] Add shape information to input node.

---
 validate_model.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/validate_model.sh b/validate_model.sh
index 2c7f59a4..1cfa30d1 100755
--- a/validate_model.sh
+++ b/validate_model.sh
@@ -143,9 +143,9 @@ bazel-bin/lib/python/tools/tf_converter --input=${TF_MODEL_FILE_PATH} \
                                         --template=${LIBMACE_SOURCE_DIR}/lib/python/tools/model.template \
                                         --model_tag=${MODEL_TAG} \
                                         ${DSP_MODE_FLAG} \
+                                        --input_shape="${INPUT_SHAPE}" \
                                         --obfuscate=True || exit -1
 
-
 echo "Step 3: Download mace static library"
 download_and_link_lib
 
-- 
GitLab