catch up to the latest interface of nerv.Networkalpha-3.4

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