• 实验环境
    1. pytorch 0.4
    2. Python 3.6
  • 在使用交叉验证后,将保存的模型重新加载求平均的结果。但是出现了以下的错误。
  • Pytorch:Unexpected key(s) in state_dict:
    实验的bug截图

    在此感谢这篇博文:https://blog.csdn.net/jacke121/article/details/82495184

  • 自己修改后的关键代码如下:

model_resnet101 = get_net()
model_resnet101.cuda()
model_resnet101.load_state_dict({k.replace('module.',''):v for k,v in torch.load("densenet169_rnn_fold_1_model_best_f1.pth.tar")['state_dict'].items()})
  • 如有错误,多谢指正。

 

相关文章:

  • 2022-12-23
  • 2021-04-23
  • 2022-01-21
  • 2021-10-02
  • 2021-11-05
  • 2021-08-30
  • 2022-02-08
  • 2021-12-29
猜你喜欢
  • 2021-11-02
  • 2022-12-23
  • 2021-09-30
  • 2021-07-15
  • 2021-04-17
  • 2022-12-23
  • 2022-12-23
相关资源
相似解决方案