Adding warm up for Zipformer2 (#766)

Signed-off-by: manickavela1998@gmail.com <manickavela1998@gmail.com>
2024-04-16 06:46:55 +05:30
parent 5981adf454
commit fb4aee83ac
9 changed files with 99 additions and 6 deletions
--- a/sherpa-onnx/csrc/online-recognizer-impl.h
+++ b/sherpa-onnx/csrc/online-recognizer-impl.h
@@ -37,6 +37,12 @@ class OnlineRecognizerImpl {

  virtual bool IsReady(OnlineStream *s) const = 0;

+  virtual void WarmpUpRecognizer(int32_t warmup, int32_t mbs) const {
+    // ToDo extending to other  models
+    SHERPA_ONNX_LOGE("Only zipformer2 model supports Warm up for now.");
+    exit(-1);
+  }
+
  virtual void DecodeStreams(OnlineStream **ss, int32_t n) const = 0;

  virtual OnlineRecognizerResult GetResult(OnlineStream *s) const = 0;