pytorch Missing key(s) in state_dict-CFANZ编程社区

pytorch Missing key(s) in state_dict

q松_松q

阅读 81

2022-02-11

加载模型的时候，加上一个关键字 strict


model.load_state_dict(pretrained_dict, strict=False)    # 多GPU训练，单GPU推理 strict=False

相关推荐
残北
 RuntimeError: Error(s) in loading state_dic ，Missing key(s) in state_dict , Unexpected key(s)
残北 119 0 0
40dba2f2a596
 加载模型出现-RuntimeError: Error(s) in loading state_dict for Net:unexpected key(s) in state_dict: XXX
40dba2f2a596 79 0 0
技术只适用于干活
 Unexpected key(s) in state_dict: “embeddings.position_ids“.
技术只适用于干活 95 0 0
丹柯yx
 RuntimeError: Error(s) in loading state_dict for
丹柯yx 77 0 0
ivy吖
 Unexpected key(s) in state_dict: "module.backbone.bn1.num_batches_tracked"
ivy吖 40 0 0
_LEON_
 Missing key(s) in state_dict: “fc4.weight“, “fc4.bias“, “fc5.weight“, “fc5.bias“
_LEON_ 10 0 0
古得曼_63b6
 【Error】Pytorch测试神经网络时出现“RuntimeError: Error(s) in loading state_dict for *** “
古得曼_63b6 77 0 0
哈哈我是你爹呀
 查看模型——model.state_dict().keys()
哈哈我是你爹呀 98 0 0
古得曼_63b6
 pytorch中model.parameters()和model.state_dict()使用时的区别
古得曼_63b6 149 0 0
飞空之羽
 Python中dict支持多个key的方法
飞空之羽 24 0 0

精彩评论（0）