refactor kaldi_decode

1 files changed, 6 insertions, 6 deletions

diff --git a/kaldi_decode/src/nnet-forward.cc b/kaldi_decode/src/nnet-forward.cc
index 4911791..8781705 100644
--- a/kaldi_decode/src/nnet-forward.cc
+++ b/kaldi_decode/src/nnet-forward.cc
@@ -46,7 +46,7 @@ int main(int argc, char *argv[]) {
         const char *usage =
             "Perform forward pass through Neural Network.\n"
             "\n"
-            "Usage:  nnet-forward [options] <nerv-config> <feature-rspecifier> <feature-wspecifier> [nerv4decode.lua]\n"
+            "Usage:  nnet-forward [options] <nerv-config> <feature-rspecifier> <feature-wspecifier> [asr_propagator.lua]\n"
             "e.g.: \n"
             " nnet-forward config.lua ark:features.ark ark:mlpoutput.ark\n";
 
@@ -78,9 +78,9 @@ int main(int argc, char *argv[]) {
         std::string config = po.GetArg(1),
             feature_rspecifier = po.GetArg(2),
             feature_wspecifier = po.GetArg(3),
-            nerv4decode = "src/nerv4decode.lua";
-        if(po.NumArgs() >= 4)
-            nerv4decode = po.GetArg(4);
+            propagator = "src/asr_propagator.lua";
+            if(po.NumArgs() >= 4)
+                propagator = po.GetArg(4);
 
         //Select the GPU
 #if HAVE_CUDA==1
@@ -99,8 +99,8 @@ int main(int argc, char *argv[]) {
 
         lua_State *L = lua_open();
         luaL_openlibs(L);
-        if(luaL_loadfile(L, nerv4decode.c_str()))
-            KALDI_ERR << "luaL_loadfile() " << nerv4decode << " failed " << lua_tostring(L, -1);
+        if(luaL_loadfile(L, propagator.c_str()))
+            KALDI_ERR << "luaL_loadfile() " << propagator << " failed " << lua_tostring(L, -1);
 
         if(lua_pcall(L, 0, 0, 0))
             KALDI_ERR << "lua_pall failed " << lua_tostring(L, -1);