[2021-04-15 15:07:55,492] [ INFO] - Found /home/aistudio/.paddlenlp/models/ernie-tiny/dict.wordseg.pickle
---------------------------------------------------------------------------KeyError Traceback (most recent call last) in
17 load_checkpoint='./ckpt/best_model/model.pdparams',
18 label_map=label_map)
---> 19 results = model.predict(data, max_seq_len=128, batch_size=1, use_gpu=False)
20 for idx, text in enumerate(data):
21 print('Data: {} \t Lable: {}'.format(text[0], results[idx]))
/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddlehub/module/nlp_module.py in predict(self, data, max_seq_len, batch_size, use_gpu)
530 results = []
531 self.eval()
--> 532 for batch in batches:
533 input_ids, segment_ids = batch
534 input_ids = paddle.to_tensor(input_ids)
/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddlehub/module/nlp_module.py in _batchify(self, data, max_seq_len, batch_size)
433 for text in data:
434 encoded_inputs = self._convert_text_to_input(tokenizer, text, max_seq_len)
--> 435 examples.append((encoded_inputs['input_ids'], encoded_inputs['segment_ids']))
436
437 # Seperates data into some batches.
KeyError: 'segment_ids'
encoded_inputs['segment_ids'] 没有这segment_ids个键值,是不是数据处理的问题