HarderThenHarder / transformers_tasks

⭐️ NLP Algorithms with transformers lib. Supporting Text-Classification, Text-Generation, Information-Extraction, Text-Matching, RLHF, SFT etc.
https://www.zhihu.com/column/c_1451236880973426688
2.11k stars 376 forks source link

运行UIE train.sh时报错,求解答 #85

Closed keyilizzie closed 7 months ago

keyilizzie commented 7 months ago

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