Add Streaming zipformer (#50)

This commit is contained in:
Fangjun Kuang
2023-02-21 20:00:03 +08:00
committed by GitHub
parent 8d1be0945e
commit 3ea6aa949d
20 changed files with 1576 additions and 99 deletions

View File

@@ -1,6 +1,7 @@
include_directories(${CMAKE_SOURCE_DIR})
add_library(sherpa-onnx-core
cat.cc
features.cc
online-lstm-transducer-model.cc
online-recognizer.cc
@@ -8,8 +9,11 @@ add_library(sherpa-onnx-core
online-transducer-greedy-search-decoder.cc
online-transducer-model-config.cc
online-transducer-model.cc
online-zipformer-transducer-model.cc
onnx-utils.cc
symbol-table.cc
text-utils.cc
unbind.cc
wave-reader.cc
)
@@ -27,3 +31,32 @@ endif()
install(TARGETS sherpa-onnx-core DESTINATION lib)
install(TARGETS sherpa-onnx DESTINATION bin)
if(SHERPA_ONNX_ENABLE_TESTS)
set(sherpa_onnx_test_srcs
cat-test.cc
unbind-test.cc
)
function(sherpa_onnx_add_test source)
get_filename_component(name ${source} NAME_WE)
set(target_name ${name})
add_executable(${target_name} "${source}")
target_link_libraries(${target_name}
PRIVATE
gtest
gtest_main
sherpa-onnx-core
)
add_test(NAME "${target_name}"
COMMAND
$<TARGET_FILE:${target_name}>
)
endfunction()
foreach(source IN LISTS sherpa_onnx_test_srcs)
sherpa_onnx_add_test(${source})
endforeach()
endif()