From 17597f452656da75c673f72ec978d332cc1ba39b Mon Sep 17 00:00:00 2001 From: Reuben Morais Date: Sun, 12 Jan 2020 15:08:32 +0100 Subject: [PATCH] Fix benchmark_model test --- taskcluster/tc-lite_benchmark_model-ds-tests.sh | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/taskcluster/tc-lite_benchmark_model-ds-tests.sh b/taskcluster/tc-lite_benchmark_model-ds-tests.sh index 580e4a1d..662e2162 100755 --- a/taskcluster/tc-lite_benchmark_model-ds-tests.sh +++ b/taskcluster/tc-lite_benchmark_model-ds-tests.sh @@ -14,7 +14,7 @@ export PATH=${TASKCLUSTER_TMP_DIR}/ds/:$PATH lite_benchmark_model \ --graph=${TASKCLUSTER_TMP_DIR}/ds/${model_name} \ --show_flops \ - --input_layer=input_node,previous_state_c,previous_state_h \ - --input_layer_type=float,float,float \ - --input_layer_shape=1,16,19,26:1:1,494:1,494 \ - --output_layer=logits,new_state_c,new_state_h + --input_layer=input_node,previous_state_c,previous_state_h,input_samples \ + --input_layer_type=float,float,float,float \ + --input_layer_shape=1,16,19,26:1,100:1,100:512 \ + --output_layer=logits,new_state_c,new_state_h,mfccs