aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorYimmon Zhuang <[email protected]>2015-10-14 15:33:48 +0800
committerYimmon Zhuang <[email protected]>2015-10-14 15:33:48 +0800
commit619cd9f2d27fa06dd6de774c8970c2eaa2f889c6 (patch)
treeb051283f6e489268a8f51754126b7c424c200b6e
parent473eb9c082224be19f147697ba951ae5bac4b4b4 (diff)
support kaldi decoder
-rw-r--r--nerv/examples/mpe_chime3.lua9
1 files changed, 5 insertions, 4 deletions
diff --git a/nerv/examples/mpe_chime3.lua b/nerv/examples/mpe_chime3.lua
index ec095b0..0615ddc 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)