>> 深度学习
hello-java-maker
为了我们,做最好的自己
展开
-
使用pytorch的相关问题总结
本文简单记录一下使用pytorch训练的时候的一些问题或者总结1、数据预处理问题https://blog.csdn.net/pengchengliu/article/details/1086687932、pytorch的使用demohttps://zhuanlan.zhihu.com/p/255723303、保存或者加载模型https://zhuanlan.zhihu.com/p/82038049...原创 2021-03-03 14:50:02 · 538 阅读 · 2 评论 -
GAN网络生成的图片太亮、不够清晰的问题解决[torchvision.utils.save_image的range]
问题最近在跑GAN网络,生成一些图片,但是生成的图片不尽如人意,质量不高,太亮,而且太模糊,但是能够保证的是轮廓和细节是很好的,说明改进是有作用的。生成的图片是上面这样的,问题在于太亮,而且不够清晰。解决方法1、考虑你的gan网络的输入是否正确2、考虑你保存图片的时候,设置是否正确而我的问题就在于保存图片的时候,设置的参数不正确导致的,这个问题困惑了我很久,现在解决了,记录一下。save_image(denorm(fake_images), os.path.join(self.sample_d原创 2021-03-01 16:25:09 · 5620 阅读 · 3 评论 -
使用tensorflow训练数据时遇到的问题总结
1、OP_REQUIRES failed at assign_op.h models这个问题的根源在于GPU不够用,解决方法也是各不相同,这里写一下几个共性方法。在eval文件中加入os.environ['CUDA_VISIBLE_DEVICES']='2'强制使用CPU验证降低batch_size修改tensorflow-gpu的版本,可能有效果更换网络2、Argument must be a dense tensor: range(0, 3) - got shape [3], bu原创 2020-10-11 10:56:12 · 14025 阅读 · 2 评论