From 12fb5f1f473a04fc2472ce5cc9273ba1afa6176d Mon Sep 17 00:00:00 2001 From: txh18 Date: Mon, 9 Nov 2015 18:25:41 +0800 Subject: small bug in lmseqreader --- nerv/examples/lmptb/lmptb/lmseqreader.lua | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nerv/examples/lmptb/lmptb/lmseqreader.lua b/nerv/examples/lmptb/lmptb/lmseqreader.lua index 3232b5a..d75167e 100644 --- a/nerv/examples/lmptb/lmptb/lmseqreader.lua +++ b/nerv/examples/lmptb/lmptb/lmseqreader.lua @@ -117,7 +117,7 @@ function LMReader:get_batch(feeds) st.store[st.head] = nil --sentence end is passed st.head = st.head + 1 end - if (inputs_s[j][i] == self.vocab.send_end_token) then + if (inputs_s[j][i] == self.vocab.sen_end_token) then flags[j][i] = bit.bor(flags[j][i], nerv.TNN.FC.SEQ_START) end end -- cgit v1.2.3