aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authortxh18 <[email protected]>2015-11-09 18:25:41 +0800
committertxh18 <[email protected]>2015-11-09 18:25:41 +0800
commit12fb5f1f473a04fc2472ce5cc9273ba1afa6176d (patch)
tree0cce4b26f73e00d80e72a6aaa9cec99cb5631e1c
parent83089241ecf243494537e78b8bfdf53224a055d3 (diff)
small bug in lmseqreader
-rw-r--r--nerv/examples/lmptb/lmptb/lmseqreader.lua2
1 files changed, 1 insertions, 1 deletions
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