diff options
| author | Kshitij Sisodia <kshitij.sisodia@arm.com> | 2021-05-20 11:18:53 +0100 |
|---|---|---|
| committer | Kshitij Sisodia <kshitij.sisodia@arm.com> | 2021-05-20 12:57:49 +0000 |
| commit | e12ac836d2110403475d0e8b4bdfec03a0874f6c (patch) | |
| tree | 2f4ac0a379cd2cf6d5ad9bc11c2a0c14abb740c3 /source | |
| parent | 659fcd951ac18d1ee7737a6ddf6a3ec162c73ca5 (diff) | |
| download | ml-embedded-evaluation-kit-e12ac836d2110403475d0e8b4bdfec03a0874f6c.tar.gz | |
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
Diffstat (limited to 'source')
| -rw-r--r-- | source/use_case/asr/usecase.cmake | 6 | ||||
| -rw-r--r-- | source/use_case/kws_asr/usecase.cmake | 4 |
2 files changed, 5 insertions, 5 deletions
diff --git a/source/use_case/asr/usecase.cmake b/source/use_case/asr/usecase.cmake index 3d18997..89014de 100644 --- a/source/use_case/asr/usecase.cmake +++ b/source/use_case/asr/usecase.cmake @@ -63,7 +63,7 @@ generate_audio_code(${${use_case}_FILE_PATH} ${SRC_GEN_DIR} ${INC_GEN_DIR} # Generate labels file set(${use_case}_LABELS_CPP_FILE Labels) generate_labels_code( - INPUT "${${use_case}_LABELS_TXT_FILE}" + INPUT "${${use_case}_LABELS_TXT_FILE}" DESTINATION_SRC ${SRC_GEN_DIR} DESTINATION_HDR ${INC_GEN_DIR} OUTPUT_FILENAME "${${use_case}_LABELS_CPP_FILE}" @@ -75,9 +75,9 @@ USER_OPTION(${use_case}_ACTIVATION_BUF_SZ "Activation buffer size for the chosen STRING) if (ETHOS_U55_ENABLED) - set(DEFAULT_MODEL_PATH ${DEFAULT_MODEL_DIR}/wav2letter_int8_vela_H128.tflite) + set(DEFAULT_MODEL_PATH ${DEFAULT_MODEL_DIR}/wav2letter_pruned_int8_vela_H128.tflite) else() - set(DEFAULT_MODEL_PATH ${DEFAULT_MODEL_DIR}/wav2letter_int8.tflite) + set(DEFAULT_MODEL_PATH ${DEFAULT_MODEL_DIR}/wav2letter_pruned_int8.tflite) endif() set(EXTRA_MODEL_CODE 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." |