From 3270a015a315932e2aef1c31da3b354d2c7a8d56 Mon Sep 17 00:00:00 2001 From: yejianwu Date: Fri, 19 Jan 2018 18:05:50 +0800 Subject: [PATCH] move validate_model.sh's params to model.config --- model.config | 10 ++++++++-- validate_model.sh | 11 ++--------- 2 files changed, 10 insertions(+), 11 deletions(-) diff --git a/model.config b/model.config index 0a4480d1..9891c35d 100644 --- a/model.config +++ b/model.config @@ -1,3 +1,9 @@ TF_INPUT_NODE=input -TF_OUTPUT_NODE=softmax/Reshape_1 -TF_OUTPUT_BR_NODE=GCN/br_result_2/fcn_br \ No newline at end of file +TF_OUTPUT_NODE=output +TF_OUTPUT_BR_NODE=before_softmax_node +TF_MODEL_FILE_PATH=path/to/tf_model_opt.pb +MODEL_TAG=mobile_net +INPUT_SHAPE=1,64,64,3 +OUTPUT_SHAPE=1,64,64,2 +RUNTIME=gpu +TUNING_OR_NOT=1 diff --git a/validate_model.sh b/validate_model.sh index bdbcb0ce..62325b3d 100755 --- a/validate_model.sh +++ b/validate_model.sh @@ -2,23 +2,16 @@ # Must run at root dir of mace project. set +x Usage() { - echo 'Usage: bash tools/validate_model.sh tools/model.config tf_model_path model_tag input_shape output_shape runtime[gpu/dsp] [tuning]' + echo 'Usage: bash tools/validate_model.sh tools/model.config' } -if [ $# -lt 6 ];then +if [ $# -lt 1 ];then Usage exit -1 fi source $1 -TF_MODEL_FILE_PATH=$2 -MODEL_TAG=$3 -INPUT_SHAPE=$4 -OUTPUT_SHAPE=$5 -RUNTIME=$6 -TUNING_OR_NOT=${7:-0} - if [ x"$RUNTIME" = x"dsp" ]; then DATA_TYPE="DT_UINT8" DEVICE_TYPE="HEXAGON" -- GitLab