diff --git a/convlab2/nlu/jointBERT/multiwoz/nlu.py b/convlab2/nlu/jointBERT/multiwoz/nlu.py index e900f1ecf2bbcfc1faa5db1caa7088b6c58154d9..8dd0014b1b3226b8ef83c98efc655e674c0dfe15 100755 --- a/convlab2/nlu/jointBERT/multiwoz/nlu.py +++ b/convlab2/nlu/jointBERT/multiwoz/nlu.py @@ -81,8 +81,8 @@ class BERTNLU(NLU): da = {} word_seq, tag_seq, new2ori = self.dataloader.bert_tokenize(ori_word_seq, ori_tag_seq) - word_seq = word_seq[:512] - tag_seq = tag_seq[:512] + word_seq = word_seq[:510] + tag_seq = tag_seq[:510] batch_data = [[ori_word_seq, ori_tag_seq, intents, da, context_seq, new2ori, word_seq, self.dataloader.seq_tag2id(tag_seq), self.dataloader.seq_intent2id(intents)]]