RuntimeError: Error(s) in loading state_dict for DataParallel:

【小白学习笔记】如下错误:

RuntimeError: Error(s) in loading state_dict for DataParallel:
        Missing key(s) in state_dict: "module.conv1.weight", "module.bn1.weight", "module.bn1.bias", 
        Unexpected key(s) in state_dict: "module.classifier.1.weight", "module.classifier.1.bias", 

这是由于网络参数不一致导致的,两种情况:

        1、加载预训练模型时发现前后参数不一致(模型更改了);

        2、训练完成后,用于测试时发现参数不一致(还是模型更改了,或者模型设置错了)

猜你喜欢

转载自blog.csdn.net/aizsa111/article/details/131771256