From a9825071b64f37e94ec65c4dbfddd06d068d2dcd Mon Sep 17 00:00:00 2001 From: Inki Dae Date: Wed, 8 Apr 2020 14:53:10 +0900 Subject: [PATCH] Add Caffe model support to start_profiler.sh Change-Id: I7e15adb651f7ad9065e2d9e11c89c888ab272fff Signed-off-by: Inki Dae --- start_profiler.sh | 18 ++++++++++++++---- 1 file changed, 14 insertions(+), 4 deletions(-) diff --git a/start_profiler.sh b/start_profiler.sh index e4c8103..9daa935 100644 --- a/start_profiler.sh +++ b/start_profiler.sh @@ -1,20 +1,30 @@ #!/bin/sh -CNT=`/usr/bin/inference_engine_test --gtest_list_tests | wc -l` -CNT=$(($CNT - 12)) +CNT=20 echo "Tflite model test case count = $CNT" +# TFLITE model LIST=$(seq 0 $CNT) for i in $LIST do /usr/bin/inference_engine_test --gtest_filter=Prefix/InferenceEngineTfliteTest.Inference/$i done +# Caffe model +CNT=8 + +echo "Caffe model test case count = $CNT" + +LIST=$(seq 0 $CNT) +for i in $LIST +do + /usr/bin/inference_engine_test --gtest_filter=Prefix/InferenceEngineCaffeTest.Inference/$i +done + # If you want to add new model tests then add script for it below # -# Make sure to calculate the test case count like this, -# CNT=$(($CNT - 31)) which depends on a number of test cases above. +# CNT=[a number of test cases] # LIST=$(seq 0 $CNT) # for i in $LIST # do -- 2.34.1