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

move validate_model.sh's params to model.config

上级 ff10ba59
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
......@@ -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"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册