diff --git a/validate_model.sh b/validate_model.sh index 4622b9fbc5310c0702280a7fd9fcb19b70477dd4..ae214bb70fde4f8f87267b73085e383ffbbeecad 100755 --- a/validate_model.sh +++ b/validate_model.sh @@ -22,9 +22,12 @@ if [ x"$RUNTIME" = x"dsp" ]; then DATA_TYPE="DT_UINT8" DEVICE_TYPE="HEXAGON" TF_OUTPUT_NODE=${TF_OUTPUT_BR_NODE} -else +elif [ x"$RUNTIME" = x"gpu" ]; then DATA_TYPE="DT_HALF" DEVICE_TYPE="OPENCL" +else + Usage + exit -1 fi VLOG_LEVEL=0