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阶段不需要，所以去掉就可以，或者在预训练阶段去掉model也可以。

下面是我的报错和处理结果：

解决方法：找到对应得文件注释掉即可如下图：

再运行就成功了。