Error(s) in loading state_dict for DataParallel

爱被打了一巴掌 2022-10-22 13:59 68阅读 0赞

写在前面

前提是一开始模型训练在单卡上,想转到多卡,resume 模型报错没有 module 模块

没有几行的正式开始

参考这个

在加载模型的地方修改一下:

  1. state_dict =checkpoint['state_dict']
  2. from collections import OrderedDict
  3. new_state_dict = OrderedDict()
  4. for k, v in state_dict.items():
  5. if 'module' not in k:
  6. k = 'module.'+k
  7. else:
  8. k = k.replace('features.module.', 'module.features.')
  9. new_state_dict[k]=v
  10. model.load_state_dict(new_state_dict)

发表评论

表情:
评论列表 (有 0 条评论,68人围观)

还没有评论,来说两句吧...

相关阅读