From 570c659ffeacd9650301310a759ffe04edafb736 Mon Sep 17 00:00:00 2001 From: wuchenghui Date: Tue, 23 Jan 2018 16:49:30 +0800 Subject: [PATCH] add softmax support for dsp --- model.config | 1 - validate_model.sh | 1 - 2 files changed, 2 deletions(-) diff --git a/model.config b/model.config index 9891c35d..a1cdf700 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 62325b3d..b20b7b56 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" -- GitLab