趁着程序还在跑,来把过程中遇到的问题总结一下——希望世间再无bug
文章目录
问题一:各个模块的安装
用到的之前没用过的模块当然需要自己安装一下,这也不算是问题比较麻烦的有两个:
一个是是安装cv2,具体方法前几天也有总结,指路——no module named cv2,但这个的方法真的有点玄学,有的方法有的人就可以,但有的就不行,所以有点耐心,把能找到的方法都试一遍,总有一个适合你。
另一个是解决pytorch_colors的问题,一开始我并没有意识到问题的严重性(毕竟第一次见),以为它也只是一个平平无奇的pip install即可解决的模块,直到倒腾了一会才意识到这原来他就是python的第三方库,然后也是各种百度,实话说这真的太能搞心态了,各路大佬可能见怪不怪所以几乎所有的方法都是从第二步开始介绍,没有人跟我说需要先下载好已经写好的含有setup.py的pytorch_colors的zip包(哭会……),后来知道真相的我,被自己蠢到在床上打滚。。。具体方法当场我就总结了,指路——如何解决PyTorch中的No module named ‘pytorch_colors‘问题
问题二:训练集样本数一直为0
出现这种问题一般是两个原因,一是图片路径出现问题,而是参数设置错误。而我就是路径问题,主要原因是没有分清python中相对路径和绝对路径。
问题三:cuda内存不足
两个方法,要么减少训练集数目,要么令batch_size=1
问题四: torch.nn.utils.clip_grad_norm函数被弃用
e:/pytnon_list/Zero-DCE-master/Zero-DCE_code/lowlight_train.py:72:
UserWarning: torch.nn.utils.clip_grad_norm is now deprecated in favor of
torch.nn.utils.clip_grad_norm_.
torch.nn.utils.clip_grad_norm(DCE_net.parameters(),config.grad_clip_norm)
在训练程序中用到了这个函数,但最后报错说它已被弃用,只用把它改成torch.nn.utils.clip_grad_norm_即可,具体代码如下:
optimizer

最低0.47元/天 解锁文章
381

被折叠的 条评论
为什么被折叠?



