From e12ac836d2110403475d0e8b4bdfec03a0874f6c Mon Sep 17 00:00:00 2001 From: Kshitij Sisodia Date: Thu, 20 May 2021 11:18:53 +0100 Subject: MLECO-1883: Updating wav2letter model Using the new pruned wav2letter model from Arm Model Zoo. The new model when optimised by Vela, produces a tflite file ~10 MB smaller than the current. Change-Id: I4ab6007c5b6111f41d8097e29b2af6cde2abc457 --- source/use_case/kws_asr/usecase.cmake | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'source/use_case/kws_asr/usecase.cmake') diff --git a/source/use_case/kws_asr/usecase.cmake b/source/use_case/kws_asr/usecase.cmake index 9ff4fff..c47d633 100644 --- a/source/use_case/kws_asr/usecase.cmake +++ b/source/use_case/kws_asr/usecase.cmake @@ -68,10 +68,10 @@ USER_OPTION(${use_case}_MODEL_SCORE_THRESHOLD_ASR "Specify the score threshold [ if (ETHOS_U55_ENABLED) set(DEFAULT_MODEL_PATH_KWS ${DEFAULT_MODEL_DIR}/ds_cnn_clustered_int8_vela_H128.tflite) - set(DEFAULT_MODEL_PATH_ASR ${DEFAULT_MODEL_DIR}/wav2letter_int8_vela_H128.tflite) + set(DEFAULT_MODEL_PATH_ASR ${DEFAULT_MODEL_DIR}/wav2letter_pruned_int8_vela_H128.tflite) else() set(DEFAULT_MODEL_PATH_KWS ${DEFAULT_MODEL_DIR}/ds_cnn_clustered_int8.tflite) - set(DEFAULT_MODEL_PATH_ASR ${DEFAULT_MODEL_DIR}/wav2letter_int8.tflite) + set(DEFAULT_MODEL_PATH_ASR ${DEFAULT_MODEL_DIR}/wav2letter_pruned_int8.tflite) endif() USER_OPTION(${use_case}_MODEL_TFLITE_PATH_KWS "NN models file to be used for KWS in the evaluation application. Model files must be in tflite format." -- cgit v1.2.1