From 6acbb9c1b0bd9ecba0a98710bdb55831ee523218 Mon Sep 17 00:00:00 2001
From: yejianwu <yejianwu@xiaomi.com>
Date: Fri, 2 Mar 2018 11:29:34 +0800
Subject: [PATCH] add vlog_level to config file

---
 env.sh        | 1 -
 example.yaml  | 2 +-
 mace_tools.py | 1 +
 3 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/env.sh b/env.sh
index 6fe87621..9e7896ce 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 69e5cb78..1d1523b0 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 d03ac326..9533df87 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 = []
-- 
GitLab