From dc7debf316a7a8bf943ff02db71c026710b3f538 Mon Sep 17 00:00:00 2001 From: liuqi Date: Wed, 31 Jan 2018 16:53:05 +0800 Subject: [PATCH] Fix some typo. --- local_validate_model.sh | 28 ++++++++++++++-------------- 1 file changed, 14 insertions(+), 14 deletions(-) diff --git a/local_validate_model.sh b/local_validate_model.sh index 1c2eeb1a..8a9d50ba 100755 --- a/local_validate_model.sh +++ b/local_validate_model.sh @@ -89,20 +89,20 @@ python tools/validate.py --generate_data true \ --input_shape="${INPUT_SHAPE}" || exit -1 echo "Step 2: Convert tf model to mace model and optimize memory" -#bazel build //lib/python/tools:tf_converter || exit -1 -#rm -rf ${MODEL_CODEGEN_DIR} -#mkdir -p ${MODEL_CODEGEN_DIR} -#bazel-bin/lib/python/tools/tf_converter --input=${TF_MODEL_FILE_PATH} \ -# --output=${MODEL_CODEGEN_DIR}/model.cc \ -# --input_node=${TF_INPUT_NODE} \ -# --output_node=${TF_OUTPUT_NODE} \ -# --data_type=${DATA_TYPE} \ -# --runtime=${RUNTIME} \ -# --output_type=source \ -# --template=${LIBMACE_SOURCE_DIR}/lib/python/tools/model.template \ -# --model_tag=${MODEL_TAG} \ -# --input_shape="${INPUT_SHAPE}" \ -# --obfuscate=False || exit -1 +bazel build //lib/python/tools:tf_converter || exit -1 +rm -rf ${MODEL_CODEGEN_DIR} +mkdir -p ${MODEL_CODEGEN_DIR} +bazel-bin/lib/python/tools/tf_converter --input=${TF_MODEL_FILE_PATH} \ + --output=${MODEL_CODEGEN_DIR}/model.cc \ + --input_node=${TF_INPUT_NODE} \ + --output_node=${TF_OUTPUT_NODE} \ + --data_type=${DATA_TYPE} \ + --runtime=${RUNTIME} \ + --output_type=source \ + --template=${LIBMACE_SOURCE_DIR}/lib/python/tools/model.template \ + --model_tag=${MODEL_TAG} \ + --input_shape="${INPUT_SHAPE}" \ + --obfuscate=False || exit -1 echo "Step 3: Download mace static library" download_and_link_lib -- GitLab