aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--nerv/examples/mmi_chime3.lua1
-rw-r--r--nerv/examples/mpe_chime3.lua10
2 files changed, 7 insertions, 4 deletions
diff --git a/nerv/examples/mmi_chime3.lua b/nerv/examples/mmi_chime3.lua
index 6ac7f28..3daaafa 100644
--- a/nerv/examples/mmi_chime3.lua
+++ b/nerv/examples/mmi_chime3.lua
@@ -160,6 +160,7 @@ function make_readers(feature_rspecifier, layer_repo)
feature_rspecifier = feature_rspecifier,
frm_ext = gconf.frm_ext,
global_transf = layer_repo:get_layer("global_transf"),
+ need_key = true,
mlfs = {}
})
}
diff --git a/nerv/examples/mpe_chime3.lua b/nerv/examples/mpe_chime3.lua
index ec095b0..f9a2855 100644
--- a/nerv/examples/mpe_chime3.lua
+++ b/nerv/examples/mpe_chime3.lua
@@ -7,6 +7,8 @@ gconf = {lrate = 0.00001, wcost = 0, momentum = 0.0,
tr_scp = "ark,s,cs:/slfs6/users/ymz09/kaldi/src/featbin/copy-feats scp:/slfs5/users/ymz09/chime/baseline/ASR/exp/tri4a_dnn_tr05_multi_enhanced_smbr/train.scp ark:- |",
initialized_param = {"/slfs6/users/ymz09/nerv-project/nerv/nerv-speech/kaldi_seq/test/chime3_init.nerv",
"/slfs6/users/ymz09/nerv-project/nerv/nerv-speech/kaldi_seq/test/chime3_global_transf.nerv"},
+ decode_param = {"/slfs6/users/ymz09/nerv-project/test_mpe/1.nerv",
+ "/slfs6/users/ymz09/nerv-project/nerv/nerv-speech/kaldi_seq/test/chime3_global_transf.nerv"},
debug = false}
function make_layer_repo(param_repo)
@@ -125,13 +127,12 @@ function make_layer_repo(param_repo)
["mpe_crit[1]"] = "<output>[1]"
}
}},
- softmax_output = {{}, {
+ decode_output = {{}, {
dim_in = {440}, dim_out = {2011},
sub_layers = layer_repo,
connections = {
["<input>[1]"] = "main[1]",
- ["main[1]"] = "softmax[1]",
- ["softmax[1]"] = "<output>[1]"
+ ["main[1]"] = "<output>[1]"
}
}}
}
@@ -145,7 +146,7 @@ function get_network(layer_repo)
end
function get_decode_network(layer_repo)
- return layer_repo:get_layer("softmax_output")
+ return layer_repo:get_layer("decode_output")
end
function get_global_transf(layer_repo)
@@ -160,6 +161,7 @@ function make_readers(feature_rspecifier, layer_repo)
feature_rspecifier = feature_rspecifier,
frm_ext = gconf.frm_ext,
global_transf = layer_repo:get_layer("global_transf"),
+ need_key = true,
mlfs = {}
})
}