From f41594b7b45272afa858859b10fd681b03984b83 Mon Sep 17 00:00:00 2001
From: Mickael Rouvier <mickael.rouvier@gmail.com>
Date: Wed, 4 Mar 2015 13:18:46 +0100
Subject: [PATCH] option for adaptation state
---
src/online2bin/online2-wav-nnet2-latgen-faster.cc | 12 ++++++++++--
1 file changed, 10 insertions(+), 2 deletions(-)
diff --git a/src/online2bin/online2-wav-nnet2-latgen-faster.cc b/src/online2bin/online2-wav-nnet2-latgen-faster.cc
index 97c9fdf..af5bd8e 100644
--- a/src/online2bin/online2-wav-nnet2-latgen-faster.cc
+++ b/src/online2bin/online2-wav-nnet2-latgen-faster.cc
@@ -175,11 +175,11 @@ int main(int argc, char *argv[]) {
OnlineTimingStats timing_stats;
+ OnlineIvectorExtractorAdaptationState adaptation_state(feature_info.ivector_extractor_info);
+
for (; !spk2utt_reader.Done(); spk2utt_reader.Next()) {
std::string spk = spk2utt_reader.Key();
const std::vector<std::string> &uttlist = spk2utt_reader.Value();
- OnlineIvectorExtractorAdaptationState adaptation_state(
- feature_info.ivector_extractor_info);
for (size_t i = 0; i < uttlist.size(); i++) {
std::string utt = uttlist[i];
if (!wav_reader.HasKey(utt)) {
@@ -193,6 +193,7 @@ int main(int argc, char *argv[]) {
SubVector<BaseFloat> data(wave_data.Data(), 0);
OnlineNnet2FeaturePipeline feature_pipeline(feature_info);
+ //ReadKaldiObject("final_adaptation_state.txt", &adaptation_state);
feature_pipeline.SetAdaptationState(adaptation_state);
SingleUtteranceNnet2Decoder decoder(nnet2_decoding_config,
@@ -255,7 +256,14 @@ int main(int argc, char *argv[]) {
KALDI_LOG << "Decoded utterance " << utt;
num_done++;
}
+ cerr<<"SAVE adaptation state"<<endl;
+ WriteKaldiObject(adaptation_state, "adaptation_state.txt", false);
+
}
+
+ //cerr<<"SAVE FINAL adaptation state"<<endl;
+ //WriteKaldiObject(adaptation_state, "final_adaptation_state.txt", false);
+
timing_stats.Print(online);
KALDI_LOG << "Decoded " << num_done << " utterances, "
--
GitLab