search

nlpyang / PreSumm

code for EMNLP 2019 paper Text Summarization with Pretrained Encoders
MIT License
1.29k stars 465 forks source link

use test_text mode, but load_state_dict wrong! #202

Closed MrZhangmc closed 3 years ago

MrZhangmc commented 3 years ago

hi, when i use test_text mode, but the load_state_dict happened wrong. Traceback (most recent call last): File "/Users/zhangmc/software_kg/PreSumm-dev/src/train.py", line 140, in <module> test_text_abs(args) File "/Users/zhangmc/software_kg/PreSumm-dev/src/train_abstractive.py", line 324, in test_text_abs model = AbsSummarizer(args, device, checkpoint) File "/Users/zhangmc/software_kg/PreSumm-dev/src/models/model_builder.py", line 217, in __init__ self.load_state_dict(checkpoint['model'], strict=True) File "/Users/zhangmc/software_kg/venv/lib/python3.7/site-packages/torch/nn/modules/module.py", line 830, in load_state_dict self.__class__.__name__, "\n\t".join(error_msgs))) RuntimeError: Error(s) in loading state_dict for AbsSummarizer: Missing key(s) in state_dict: "bert.model.embeddings.position_ids". can you give me help, thank you!

SinaSarparast commented 2 years ago

@MrZhangmc Did you manage to solve this issue? I also have the same issue.