06 2019 档案

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 阅读(5346) 评论(0) 推荐(0) 编辑
pytorch--cpu与gpu load时相互转化
摘要：pytorch cpu与gpu load时相互转化 torch.load(map_location=)学习 将gpu改为cpu时，遇到一个报错：RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda 阅读全文
posted @ 2019-06-06 18:18 U_C 阅读(2851) 评论(0) 推荐(1) 编辑
pytorch 调用forward 的具体流程
摘要：forward方法的具体流程： 以一个Module为例：1. 调用module的call方法2. module的call里面调用module的forward方法3. forward里面如果碰到Module的子类，回到第1步，如果碰到的是Function的子类，继续往下4. 调用Function的ca 阅读全文
posted @ 2019-06-03 15:04 U_C 阅读(6164) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示