experiments/STT
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix benchmark_model test

Browse Source
This commit is contained in:
Reuben Morais 2020-01-12 15:08:32 +01:00
parent cce1cec740
commit 17597f4526
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
taskcluster/tc-lite_benchmark_model-ds-tests.sh
View File

@ -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