From 0b5dbd20b22635398c3a13ddf00bda1930a0ef44 Mon Sep 17 00:00:00 2001 From: Liangliang He Date: Thu, 18 Jan 2018 18:12:51 +0800 Subject: [PATCH] Add activations support --- validate_model.sh | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/validate_model.sh b/validate_model.sh index 4622b9fb..ae214bb7 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 -- GitLab