Traceback (most recent call last):
File "/home/liukeyi/transformers_tasks/UIE/train.py", line 328, in
train()
File "/home/liukeyi/transformers_tasks/UIE/train.py", line 249, in train
model = torch.load(os.path.join(args.pretrained_model, '/home/liukeyi/uie-base-zh/pytorch_model.bin')) # 加载预训练好的UIE模型,模型结构见:model.UIE()
File "/home/liukeyi/miniconda3/envs/ttuie/lib/python3.9/site-packages/torch/serialization.py", line 1028, in load
return _legacy_load(opened_file, map_location, pickle_module, pickle_load_args)
File "/home/liukeyi/miniconda3/envs/ttuie/lib/python3.9/site-packages/torch/serialization.py", line 1246, in _legacy_load
magic_number = pickle_module.load(f, pickle_load_args)
EOFError: Ran out of input
Traceback (most recent call last): File "/home/liukeyi/transformers_tasks/UIE/train.py", line 328, in
train()
File "/home/liukeyi/transformers_tasks/UIE/train.py", line 249, in train
model = torch.load(os.path.join(args.pretrained_model, '/home/liukeyi/uie-base-zh/pytorch_model.bin')) # 加载预训练好的UIE模型,模型结构见:model.UIE()
File "/home/liukeyi/miniconda3/envs/ttuie/lib/python3.9/site-packages/torch/serialization.py", line 1028, in load
return _legacy_load(opened_file, map_location, pickle_module, pickle_load_args)
File "/home/liukeyi/miniconda3/envs/ttuie/lib/python3.9/site-packages/torch/serialization.py", line 1246, in _legacy_load
magic_number = pickle_module.load(f, pickle_load_args)
EOFError: Ran out of input