diff --git a/model.config b/model.config
index 0a4480d196671271ec429f294b12d65fa4243b4b..9891c35d6a350aeec5a8f2e92160d5535bf0667c 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 bdbcb0ce8b1477a23cca4681066fb28885a6c342..62325b3d817b80b59350caf3fa2a22be7e2a10f0 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"