这种错误多半是模型加载错误导致。排查测试代码的模型,与加载的模型是否是同一个。
错误代码:Missing key(s) in state_dict: "input_block.conv1.conv.weight",... Unexpected key(s) in state_dict: "module.input_block.conv1.conv.weight",...错误分析:
区别在于一个加了module,一个没加。
解决办法:训练模型时,在服务器使用了多GPUtorch.nn.DataParallel()
测试模型时,在本地添加代码torch.nn.DataParallel()即可。



