diff --git a/model.config b/model.config index b9b75e95e926d8df0f68f2ce5c729cf6b4999a3b..ccfc11591e53090c6e7abaa9a7f8ba4a4079cbd1 100644 --- a/model.config +++ b/model.config @@ -8,3 +8,4 @@ RUNTIME=gpu TUNING_OR_NOT=1 ANDROID_ABI=armeabi-v7a # Or 'arm64-v8a' LIMIT_OPENCL_KERNEL_TIME=0 +DSP_MODE=0 # used only for dsp diff --git a/validate_model.sh b/validate_model.sh index b8cfdaf4c7322df73bdae359f715762cf0952daa..6ee71ec2d190bc2b66cb128c6786583951c549d3 100755 --- a/validate_model.sh +++ b/validate_model.sh @@ -134,6 +134,9 @@ 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} +if [ ${DSP_MODE} ]; then + DSP_MODE_FLAG="--dsp_mode=${DSP_MODE}" +fi bazel-bin/lib/python/tools/tf_converter --input=${TF_MODEL_FILE_PATH} \ --output=${MODEL_CODEGEN_DIR}/model.cc \ --input_node=${TF_INPUT_NODE} \ @@ -143,6 +146,7 @@ bazel-bin/lib/python/tools/tf_converter --input=${TF_MODEL_FILE_PATH} \ --output_type=source \ --template=${LIBMACE_SOURCE_DIR}/lib/python/tools/model.template \ --model_tag=${MODEL_TAG} \ + ${DSP_MODE_FLAG} \ --obfuscate=True || exit -1 echo "Step 3: Download mace static library"