diff options
Diffstat (limited to 'kaldi_decode')
| -rw-r--r-- | kaldi_decode/src/asr_propagator.lua | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/kaldi_decode/src/asr_propagator.lua b/kaldi_decode/src/asr_propagator.lua index 6a95647..ff9b8a2 100644 --- a/kaldi_decode/src/asr_propagator.lua +++ b/kaldi_decode/src/asr_propagator.lua @@ -61,14 +61,14 @@ function build_propagator(ifname, feature) else transformed = data[id] end - table.insert(input, transformed) + table.insert(input, {transformed}) end - local output = {nerv.MMatrixFloat(input[1]:nrow(), network.dim_out[1])} + local output = {{nerv.MMatrixFloat(input[1][1]:nrow(), network.dim_out[1])}} network:mini_batch_init({seq_length = table.vector(gconf.batch_size, 1), new_seq = {}, do_train = false, - input = {input}, - output = {output}, + input = input, + output = output, err_input = {}, err_output = {}}) network:propagate() @@ -79,7 +79,7 @@ function build_propagator(ifname, feature) end collectgarbage("collect") - return utt, output[1] + return utt, output[1][1] end return batch_propagator |