TypeError: load_state_dict() missing 1 required positional argument: ‘state_dict‘

记录最近遇到的小bug,希望能够帮助到和我有类似错误的你。

1.TypeError: parameters() missing 1 required positional argument: ‘self’

这里是用如下的方法查看model的参数量:

n_parameters = sum(p.numel() for p in model.parameters() if p.requires_grad)
print('number of params:', n_parameters)

完整报错:

Traceback (most recent call last):
  File "train_extract_feature.py", line 283, in <module>
    n_parameters = sum(p.numel() for p in model.parameters() if p.requires_grad)
TypeError: parameters() missing 1 required positional argument: 'self'

报错原因:
出现这个问题的原因是,使用这个方法时,model没有进行实例化。在我的项目中是没有执行model.load_state_dict(new_dict, strict=False)这个函数,就调用了model.parameters()的方法了,所以会出错。应该先load进这个model的weight,然后再使用parameters().

2.TypeError: load_state_dict() missing 1 required positional argument: ‘state_dict’

在使用如下命令语句报错:

 model.load_state_dict(new_dict, strict=False)

完整报错:

Traceback (most recent call last):
  File "train_extract_feature.py", line 322, in <module>
    model.load_state_dict(new_dict, strict=False)
TypeError: load_state_dict() missing 1 required positional argument: 'state_dict'

这是一个我认为很奇怪的报错内容。一开始觉得我明明有state_dict,为什么一直报这个错误呢,接着就想到可能是我的state_dict类型有问题。接下来就依次排查。
首先,检查new_dict是不是dict类型。不同的pre-trained model打包方式是不同的,像我这次的,如果查看new_dict.keys()会发现它是三个keys,epoch,name(记不清是不是name),model,所以想取到model还得用字典取值的方法,如new_dict[‘model’],取出dict类型的model,此时的new_dict[‘model’]中包含的是keys和values。检查无误后还是报相同的错误。
然后,突然想到会不会是因为model的类型不正确导致找不到load_state_dict()方法,于是我比较了能够正常运行的程序和不能正常运行的程序,然后发现model的类型果然是不一样的,心里那个激动呀。。。接着跳转到model定义的地方,发现问题出在继承类,有两个参数是空值。因为在父类中有定义了,因此没有报找不到参数的错误,而是报了这么个奇怪的错误。
中间也怀疑过是不是torch版本的问题,然后将pytorch1.8换成了1.7,没能解决问题,defeat。。。
希望我解决问题的方式能够拓展大家解决问题的思路。
欢迎大家批评指正!

评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值