Add online LSTM transducer model (#25)

2023-02-18 21:35:15 +08:00
parent b2d96c1d9a
commit cb8f85ff83
28 changed files with 1315 additions and 984 deletions
--- a/sherpa-onnx/csrc/online-transducer-model-config.h
+++ b/sherpa-onnx/csrc/online-transducer-model-config.h
@@ -0,0 +1,23 @@
+// sherpa/csrc/online-transducer-model-config.h
+//
+// Copyright (c)  2023  Xiaomi Corporation
+#ifndef SHERPA_ONNX_CSRC_ONLINE_TRANSDUCER_MODEL_CONFIG_H_
+#define SHERPA_ONNX_CSRC_ONLINE_TRANSDUCER_MODEL_CONFIG_H_
+
+#include <string>
+
+namespace sherpa_onnx {
+
+struct OnlineTransducerModelConfig {
+  std::string encoder_filename;
+  std::string decoder_filename;
+  std::string joiner_filename;
+  int32_t num_threads;
+  bool debug = false;
+
+  std::string ToString() const;
+};
+
+}  // namespace sherpa_onnx
+
+#endif  // SHERPA_ONNX_CSRC_ONLINE_TRANSDUCER_MODEL_CONFIG_H_