提交 3270a015 编写于 作者: Y yejianwu

move validate_model.sh's params to model.config

上级 ff10ba59
TF_INPUT_NODE=input TF_INPUT_NODE=input
TF_OUTPUT_NODE=softmax/Reshape_1 TF_OUTPUT_NODE=output
TF_OUTPUT_BR_NODE=GCN/br_result_2/fcn_br TF_OUTPUT_BR_NODE=before_softmax_node
\ No newline at end of file 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
...@@ -2,23 +2,16 @@ ...@@ -2,23 +2,16 @@
# Must run at root dir of mace project. # Must run at root dir of mace project.
set +x set +x
Usage() { 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 Usage
exit -1 exit -1
fi fi
source $1 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 if [ x"$RUNTIME" = x"dsp" ]; then
DATA_TYPE="DT_UINT8" DATA_TYPE="DT_UINT8"
DEVICE_TYPE="HEXAGON" DEVICE_TYPE="HEXAGON"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册