weixin_42416791的博客

Pytorch加载保存好的模型发现与实际保存的参数不一致发现问题初衷：问题：保存模型并打印参数：加载模型并打印模型参数问题来了解决问题先说结论原因：联想问题本质二级目录三级目录参考博客发现问题初衷：最近跑深度学习代码，数据量比较庞大所以训练速度慢的一，一旦有什么问题训练突然停止了，心态都要炸了，所以这时候使用pytorch保存每轮训练好的模型（可以使用一些保存技巧，保存得到的当前最新的模型checkpoint，倒不至于每轮都保存），即使训练停止还能接着去加载保存好的模型得到当前模型输出结果，节省时间，

问题在pytorch-gpu10.0 即cuda10.0的pytorch虚拟环境中（torch1.2.0）运行VoronoiNet代码时候，会出现上面的错误，一开始我以为是因为学姐的代码对pytorch版本有要求，但是问了学姐那里的torch版本就是1.2.0的，就很奇怪为什么我这里明明是相同的环境怎么就不行，有bug。在网上查找原因，发现基本都是说torch版本的问题 →????使用torch.load()加载模型参数时，提示“xxx.pt is a zip archive(did you mea

遇到问题最近在用pytorch跑深度学习模型，不过一直用的是CPU，训练速度还是跟不上，正好实验室服务器比较空闲，而且cuda、cudnn也都是配置好了的，所以打算在上面的anaconda虚拟环境中配置一下我想要用的gpu版本的pytorch，配置期间遇到了不少问题，也参考了很多博主的经验，现自己总结记录一下。总结先说结论，代码可以跑起来了就会发现，事情怎么这么简单（笑哭），但其实是将近一周的折腾才搞好的。因为我只想连服务器的gpu用一下，上面cuda、cudnn都已经配置好了的，我只需要装一个gpu