PyTorch加载模型时报错RuntimeError: Error(s) in loading state_dict for *****: Missing key(s) in state_dict:
问题描述:原作者的代码里面没有断点续训,我增加了这个功能,同时也引入了更多的参数,保存模型时增加了epoch、net.state_dict()、optimizer.state_dict()、scheduler.state_dict()等信息。原来的保存模型的代码如下:torch.save(net.state_dict(),model_dir)增加了信息后,保存模型的代码如下:torch.save(