diff --git a/model.config b/model.config index 9891c35d6a350aeec5a8f2e92160d5535bf0667c..a1cdf700122bffc19120ba6d2622f4f9b99301c7 100644 --- a/model.config +++ b/model.config @@ -1,6 +1,5 @@ TF_INPUT_NODE=input 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 diff --git a/validate_model.sh b/validate_model.sh index 62325b3d817b80b59350caf3fa2a22be7e2a10f0..b20b7b567ef7cd3ea3f1544b638992a5bf919f8a 100755 --- a/validate_model.sh +++ b/validate_model.sh @@ -15,7 +15,6 @@ source $1 if [ x"$RUNTIME" = x"dsp" ]; then DATA_TYPE="DT_UINT8" DEVICE_TYPE="HEXAGON" - TF_OUTPUT_NODE=${TF_OUTPUT_BR_NODE} elif [ x"$RUNTIME" = x"gpu" ]; then DATA_TYPE="DT_HALF" DEVICE_TYPE="OPENCL"