2019年6月18日
Missing key(s) in state_dict: Unexpected key(s) in state_dict
摘要： 如果加载的预训练模型之前使用了torch.nn.DataParallel()，而此时的训练并没有使用，则会出现这样的错误。解决方案有两个：1：此时的训练加入torch.nn.DataParallel()即可。2：创建一个没有module.的新字典，即将原来字典中module.删除掉。解决方案1：mo 阅读全文
posted @ 2019-06-18 14:44 U_C 阅读(5220) 评论(0) 推荐(0) 编辑