用自定义数据集训练您的模型，出错

[qiunlp]

自定义数据集格式和semeval完全一样，只是关系的种类多一些。请指教是什么原因？

ValueError: Caught ValueError in DataLoader worker process 1. Original Traceback (most recent call last): File "/home/qwh/.local/lib/python3.6/site-packages/torch/utils/data/_utils/worker.py", line 178, in _worker_loop data = fetcher.fetch(index) File "/home/qwh/.local/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "/home/qwh/.local/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 44, in data = [self.dataset[idx] for idx in possibly_batched_index] File "/home/qwh/桌面/REDN/opennre/framework/data_loader.py", line 143, in getitem seq_len) File "/home/qwh/桌面/REDN/opennre/framework/data_loader.py", line 155, in merge_ont_hot_labels em = SentenceREDataset.get_entity_mask(pos_head, pos_tail, new_index, seq_len) File "/home/qwh/桌面/REDN/opennre/framework/data_loader.py", line 166, in get_entity_mask pos_tail = [new_index.index(i) for i in pos_tail] File "/home/qwh/桌面/REDN/opennre/framework/data_loader.py", line 166, in pos_tail = [new_index.index(i) for i in pos_tail] ValueError: 7 is not in list

4%|▍ | 4/104 [00:00<00:07, 12.73it/s]