RuntimeError: Error(s) in loading state_dict for TransformerMASSModel:解决方法

运行MASS/MASS-summarization代码报错:RuntimeError: Error(s) in loading state_dict for TransformerMASSModel: 因为fine-tuning阶段需要加载之前训练好得Pre-training模型,在之前训练好得预训练模型中包含model,fine-tuning阶段不需要,所以去掉就可以,或者在预训练阶段去掉mod
相关文章
相关标签/搜索