From e6d9e562fa42ddafac601be12da4c4faee85dd4d Mon Sep 17 00:00:00 2001 From: Determinant Date: Mon, 4 Apr 2016 01:17:49 +0800 Subject: catch up to the latest interface of nerv.Network --- kaldi_decode/src/asr_propagator.lua | 10 +++++----- 1 file 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 -- cgit v1.2.3