Error(s) in loading state_dict for CNN

这个错误通常是由于尝试使用不同形状的模型参数加载模型状态字典而导致的。请检查您正在加载的状态字典是否与您尝试加载它的模型具有相同的形状。

您可以使用以下代码检查您的模型状态字典的形状：

model = YourModel()
state_dict = torch.load('your_state_dict.pth')
print(state_dict.keys()) # 查看字典中包含哪些参数
print(model.state_dict().keys()) # 查看模型中包含哪些参数

如果这两个打印出来的结果不同，那么您需要调整您的状态字典或模型以匹配它们的形状。您可以通过调整您的模型定义或使用 PyTorch 的 `nn.Module.load_state_dict()` 方法来实现这一点。例如：

model = YourModel()
state_dict = torch.load('your_state_dict.pth')
new_state_dict = {}
for k, v in state_dict.items():
    name = k.replace('module.', '') # 处理多GPU训练的情况
    new_state_dict[name] = v
model.load_state_dict(new_state_dict)

这样，您就可以将状态字典加载到您的模型中，而不会出现形状不匹配的错误。