首页 PaddleHub 帖子详情
paddlehub 在finetune时报错
收藏
快速回复
PaddleHub 问答迁移学习 856 1
paddlehub 在finetune时报错
收藏
快速回复
PaddleHub 问答迁移学习 856 1

[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'

0
收藏
回复
全部评论(1)
时间顺序
FutureSI
#2 回复于2021-12

encoded_inputs['segment_ids'] 没有这segment_ids个键值,是不是数据处理的问题

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户