【Pytorch Debug】加载保存模型 每次推理结果都不同 / unexpected_keys或者missing_keys / torch.nn.DataParallel( ).cuda()

bug描述】:模型为仅保存参数的模型,已确定加载语句 “没有问题” ,且每次测试的图片数据固定,但是每次推理得到的结果都不一样。

bug分析】:实际上在加载模型并不是完全没有问题,在使用如下加载语句时出现

cnn.load_state_dict(torch.load(os.path.join(MODEL_PATH, "model.pth"))['model'])

#保存语句如下:
#torch.save({
#    'model':cnn.state_dict(),
#    'optimizer': optimizer.state_dict()},
#    os.path.join(MODEL_PATH, "model.pth"))

会报一个:Missing key(s) in state_dict ....和 Unexpected key(s)...的错误,但是我啥也没管通过下面这种方式压了下去:

cnn.load_state_dict(torch.load(os.path.join(MODEL_PATH, "model.pth"))['model'], False)

下面我们先看看加了这个 False 对模型参数加载有什么影响。下面是load_state_dict的函数定义:

load_state_dict(self, state_dict, strict=True)

strict 的作用是判断上面参数拷贝过程中是否有unexpected_keys或者

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值