diff --git a/env.sh b/env.sh index 6fe87621fbefb6b25b9e967124886246c4a987da..9e7896ceef39d6b1137fa4c0ec702486f7df03fc 100644 --- a/env.sh +++ b/env.sh @@ -1,7 +1,6 @@ #!/usr/bin/env bash LIBMACE_TAG=`git describe --abbrev=0 --tags` -VLOG_LEVEL=0 LIBMACE_SOURCE_DIR=`/bin/pwd` INPUT_FILE_NAME="model_input" OUTPUT_FILE_NAME="model.out" diff --git a/example.yaml b/example.yaml index 69e5cb78f6e5a958b8bd2500c28a736f91d2c01c..1d1523b09b449f867cc6dc285c84b9a07a9a2874 100644 --- a/example.yaml +++ b/example.yaml @@ -3,6 +3,7 @@ # which can contains more than one models target_abi: armeabi-v7a # arm64-v8a embed_model_data: 1 +vlog_level: 0 models: preview_net: model_file_path: path/to/model64.pb @@ -22,4 +23,3 @@ models: runtime: gpu limit_opencl_kernel_time: 1 dsp_mode: 0 - diff --git a/mace_tools.py b/mace_tools.py index d03ac3263961a114f830873bbee20688020ccf94..9533df877d95c319e1ddd7c6be023cdad4a80d79 100644 --- a/mace_tools.py +++ b/mace_tools.py @@ -191,6 +191,7 @@ def main(unused_args): # Transfer params by environment os.environ["TARGET_ABI"] = target_abi os.environ["EMBED_MODEL_DATA"] = str(configs["embed_model_data"]) + os.environ["VLOG_LEVEL"] = str(configs["vlog_level"]) os.environ["PROJECT_NAME"] = os.path.splitext(FLAGS.config)[0] model_output_dirs = []