pytorch学习002- -debug(load_state_dict() missing 1 required positional argument: ‘state_dict‘)
Debug
savept.py文件用来将gpu上训练的模型转换为cpu上推理可用的pt文件
以下为部分代码:
import torch
from model import NET
print(torch.__version__) # 1.10.1+cpu
model = NET
device = torch.device('cpu')
state_dict = torch.load(PATH, map_location=device)
model.load_state_dict(state_dict, strict=False)
具体的报错内容
model.load_state_dict(state_dict, strict=False)
TypeError: load_state_dict() missing 1 required positional argument: ‘state_dict’
=打印一下state_dict:
可以看到的是正常的,一个关键在于实例化*
参考
修改:model= NET()
这个地方是属于实例化错误,以后的学习过程中应当重视基础。
补充:
在测试过程中遇到了一个“Unexpected key(s) in state_dict”的问题
解决方案:CSDN博客