transformers_tasks icon indicating copy to clipboard operation
transformers_tasks copied to clipboard

UIE信息抽取

Open xxm1668 opened this issue 2 years ago • 3 comments

结果没有你写的那么高,只有F1值只有70%左右

xxm1668 avatar Apr 04 '23 08:04 xxm1668

在运行UIE模型的train.py文件的时候出现了以下报错: Traceback (most recent call last): File "/tmp/pycharm_project_244/UIE/train.py", line 328, in train() File "/tmp/pycharm_project_244/UIE/train.py", line 249, in train model = torch.load(os.path.join(args.pretrained_model, 'pytorch_model.bin')) # 加载预训练好的UIE模型,模型结构见:model.UIE() File "/usr/local/lib/python3.8/dist-packages/torch/serialization.py", line 608, in load return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args) File "/usr/local/lib/python3.8/dist-packages/torch/serialization.py", line 777, in _legacy_load magic_number = pickle_module.load(f, **pickle_load_args) _pickle.UnpicklingError: invalid load key, '<'.

请问博主用的是什么pytorch版本?

limefilter avatar Apr 06 '23 09:04 limefilter

我也是这个情况

White-bbb avatar May 10 '23 07:05 White-bbb

+1 求解答

keyilizzie avatar Jan 14 '24 03:01 keyilizzie