犯蠢日记一-GPU上训练，加载到CPU时出问题

【解决一半】完善一下：
本来是在GPU上训练的模型，然后在CPU上加载进行测试时出现

Missing key(s) in state_dict: "conv_block1.conv_block.0.conv_3_1.weight"···
Unexpected key(s) in state_dict: "conv_block1.conv_block.0.weight"·········

然后就想试试能不能在GPU上保存模型时直接

model.cpu().state_dict()

结果训练的时候就在图一的那个位置报错如图二

图一
图二

问题补充描述：

代码中已经用了
model = model.to(device)
input = input.to(device)

**

原因及解决方法
是我在GPU上跑的时候改了模型忘记在CPU中改了，现在对应着改了后可以在cpu上加载了，但是还是不能在GPU中保存用.cpu()保存模型。
保存模型的时候直接model.state_dict()，用model.cpu().state_dict()还是不行。

**
哈哈哈真犯蠢日记