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

本文解决了在运行MASS/MASS-summarization代码时遇到的RuntimeError问题,详细介绍了错误原因及处理方法,指出fine-tuning阶段加载预训练模型时的常见误区,并提供了解决方案。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

运行MASS/MASS-summarization代码报错:RuntimeError: Error(s) in loading state_dict for TransformerMASSModel:

因为fine-tuning阶段需要加载之前训练好得Pre-training模型,在之前训练好得预训练模型中包含model,fine-tuning阶段不需要,所以去掉就可以,或者在预训练阶段去掉model也可以。

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

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

再运行就成功了。

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值