Skip to content

Commit 7740dbf

Browse files
Ebranchformer (#1951)
* adding ebranchformer encoder * extend surfaced FeatureExtractorConfig - so ebranchformer feature extraction can be configured from Python - the GlobCmvn is not needed, as it is a module in the OnnxEncoder * clean the code * Integrating remarks from Fangjun
1 parent 209eaaa commit 7740dbf

File tree

8 files changed

+609
-5
lines changed

8 files changed

+609
-5
lines changed

sherpa-onnx/csrc/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,7 @@ set(sources
6868
online-ctc-fst-decoder.cc
6969
online-ctc-greedy-search-decoder.cc
7070
online-ctc-model.cc
71+
online-ebranchformer-transducer-model.cc
7172
online-lm-config.cc
7273
online-lm.cc
7374
online-lstm-transducer-model.cc

sherpa-onnx/csrc/features.cc

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,9 @@ std::string FeatureExtractorConfig::ToString() const {
4848
os << "feature_dim=" << feature_dim << ", ";
4949
os << "low_freq=" << low_freq << ", ";
5050
os << "high_freq=" << high_freq << ", ";
51-
os << "dither=" << dither << ")";
51+
os << "dither=" << dither << ", ";
52+
os << "normalize_samples=" << (normalize_samples ? "True" : "False") << ", ";
53+
os << "snip_edges=" << (snip_edges ? "True" : "False") << ")";
5254

5355
return os.str();
5456
}

0 commit comments

Comments
 (0)