model.load_state_dict(state_dict,strict=False)

我们在构造好了一个模型后&＃xff0c;可能要加载一些训练好的模型参数。举例子如下&＃xff1a;

假设  trained.pth 是一个训练好的网络的模型参数存储

model &＃61; Net()是我们刚刚生成的一个新模型&＃xff0c;我们希望model将trained.pth中的参数加载加载进来&＃xff0c;但是model中多了一些trained.pth中不存在的参数&＃xff0c;如果使用下面的命令&＃xff1a;

state_dict &＃61; torch.load(&＃39;trained.pth&＃39;)model.load_state_dict(state_dict)

会报错&＃xff0c;说key对应不上&＃xff0c;因为model你强人所难&＃xff0c;我堂堂trained.pth没有你的那些个零碎玩意&＃xff0c;你非要向我索取&＃xff0c;我上哪给你弄去。

但是model不干&＃xff0c;说既然你不能完全满足我的需要&＃xff0c;那么你有什么我就拿什么吧&＃xff0c;怎么办呢&＃xff1f;下面的指令代码就行了。

model.load_state_dict(state_dict, strict&＃61;False)

这个博主好有趣&＃xff0c;通俗易懂
原文链接&＃xff1a;https://blog.csdn.net/t20134297/article/details/110533007