EntityRelationExtraction icon indicating copy to clipboard operation
EntityRelationExtraction copied to clipboard

训练trainer_ner报错

Open leizhu1989 opened this issue 3 years ago • 1 comments

你好,我训练的时候报了个错啊: USE_CUDA.... Run EntityRelationExtraction NER ... Traceback (most recent call last): File "mains/trainer_ner.py", line 164, in data_processor = ModelDataPreparation(config) File "/home/ctgcdt/zhulei/NLP/EntityRelationExtraction/data_loader/process_ner.py", line 27, in init self.get_type2id() File "/home/ctgcdt/zhulei/NLP/EntityRelationExtraction/data_loader/process_ner.py", line 103, in get_type2id self.token2id[line[0]] = cnt IndexError: list index out of range

字典解析错误吗,我用的是官方的vocab.txt

leizhu1989 avatar Apr 07 '23 06:04 leizhu1989

我修改了下解析vocab代码这个解决了

leizhu1989 avatar Apr 07 '23 07:04 leizhu1989