错误分析
代码块:
model.load_state_dict(torch.load(checkpoint, map_location='cpu'))
控制台报错提示:
Unexpected key(s) in state_dict: “epoch”, “model”.
或者是
Unexpected key(s) in state_dict: “XXX”, “XXX”.
该报错快速直接解决方法
model.load_state_dict 时设置 strict=False 即可解决该报错
代码为: