EntityRelationExtraction
EntityRelationExtraction copied to clipboard
实体关系抽取pipline方式,使用了BiLSTM+CRF+BERT
训练报错
您好作者,我的环境中: pytorch:1.7.0+cu110 transformer:4.3.2 pytorch-crf:0.7.2 显卡:tesla V100 训练trainer_ner的时候报错了: /pytorch/aten/src/ATen/native/cuda/Indexing.cu:658: indexSelectLargeIndex: block: [430,0,0], thread: [64,0,0] Assertion `srcIndex < srcSelectDimSize` failed. /pytorch/aten/src/ATen/native/cuda/Indexing.cu:658: indexSelectLargeIndex: block: [430,0,0], thread: [65,0,0] Assertion `srcIndex < srcSelectDimSize` failed. /pytorch/aten/src/ATen/native/cuda/Indexing.cu:658:...
你好,我训练的时候报了个错啊: 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,...
大佬,您好,感谢您的开源获益匪浅 有个问题想请教下: 就是假如我只有实体,没有客体,客体是答案做成了知识图谱那样的,我只想知道问问题的人意图去检索答案,这样的话需要怎么训练呢;比如语料“播放视频很卡是什么原因”,‘’视频没法播放啥原因‘’这样的呢,求大佬指点
调试很久始终报错
../data/vec.txt这个文件好像没有
直接运行demo.py会报错 Traceback (most recent call last): File "d:\1Code\EntityRelationExtraction\deploy\demo.py", line 120, in test() File "d:\1Code\EntityRelationExtraction\deploy\demo.py", line 72, in test ner_model_dict = torch.load(PATH_NER) File "D:\Anaconda\envs\myenv\lib\site-packages\torch\serialization.py", line 1319, in load with _open_file_like(f, "rb")...