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