bugfix (missing SetFeatureDim())

k2-fsa · Mar 14, 2024 · 96c2a9f · 96c2a9f
1 parent 468fe1c
commit 96c2a9f
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/sherpa-onnx/csrc/keyword-spotter-transducer-impl.h b/sherpa-onnx/csrc/keyword-spotter-transducer-impl.h
@@ -68,6 +68,9 @@ class KeywordSpotterTransducerImpl : public KeywordSpotterImpl {
       : config_(config),
         model_(OnlineTransducerModel::Create(config.model_config)),
         sym_(config.model_config.tokens) {
+
+    model_->SetFeatureDim(config.feat_config.feature_dim);
+
     if (sym_.contains("<unk>")) {
       unk_id_ = sym_["<unk>"];
     }
@@ -85,6 +88,9 @@ class KeywordSpotterTransducerImpl : public KeywordSpotterImpl {
       : config_(config),
         model_(OnlineTransducerModel::Create(mgr, config.model_config)),
         sym_(mgr, config.model_config.tokens) {
+
+    model_->SetFeatureDim(config.feat_config.feature_dim);
+
     if (sym_.contains("<unk>")) {
       unk_id_ = sym_["<unk>"];
     }

diff --git a/sherpa-onnx/csrc/online-recognizer-transducer-impl.h b/sherpa-onnx/csrc/online-recognizer-transducer-impl.h
@@ -14,6 +14,8 @@
 #include <utility>
 #include <vector>
 
+#include <iostream>
+
 #if __ANDROID_API__ >= 9
 #include <strstream>
 
@@ -122,6 +124,9 @@ class OnlineRecognizerTransducerImpl : public OnlineRecognizerImpl {
         model_(OnlineTransducerModel::Create(mgr, config.model_config)),
         sym_(mgr, config.model_config.tokens),
         endpoint_(config_.endpoint_config) {
+
+    model_->SetFeatureDim(config.feat_config.feature_dim);
+
     if (sym_.contains("<unk>")) {
       unk_id_ = sym_["<unk>"];
     }
@@ -153,6 +158,7 @@ class OnlineRecognizerTransducerImpl : public OnlineRecognizerImpl {
 #endif
 
   std::unique_ptr<OnlineStream> CreateStream() const override {
+    std::cout << "DEBUG, CreateStream(): " << config_.feat_config.ToString() << std::endl;
     auto stream =
         std::make_unique<OnlineStream>(config_.feat_config, hotwords_graph_);
     InitOnlineStream(stream.get());