diff options
Diffstat (limited to 'nerv/examples/lmptb/lmptb/lmseqreader.lua')
| -rw-r--r-- | nerv/examples/lmptb/lmptb/lmseqreader.lua | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/nerv/examples/lmptb/lmptb/lmseqreader.lua b/nerv/examples/lmptb/lmptb/lmseqreader.lua index f7e2539..3232b5a 100644 --- a/nerv/examples/lmptb/lmptb/lmseqreader.lua +++ b/nerv/examples/lmptb/lmptb/lmseqreader.lua @@ -105,6 +105,9 @@ function LMReader:get_batch(feeds) labels_s[j][i] = self.vocab.null_token end if (inputs_s[j][i] ~= self.vocab.null_token) then + if (labels_s[j][i] == self.vocab.null_token) then + nerv.error("reader error : label is null while input is not null") + end flags[j][i] = bit.bor(flags[j][i], nerv.TNN.FC.SEQ_NORM) got_new = true st.store[st.head] = nil |