2019.10.8 Rcan代码修改心得
今天是我修改rcan代码,想让它跑起来的第不知道多少天。因为我学过的东西总是会忘记,因此,在博客上进行记录。
今天遇到的第一个问题就是,测试集不够,这个很容易就解决了。
今天遇到的第二个问题就是在GPU上运行内存不足了,所以在代码上找,怎么才能在cpu上跑代码。在__init__.py里找到了下面的这句话
又联想了一下之前是在哪里找到训练集要执行800次的,所以在option.py里找到了
今天的第三个问题是比较巨大的。是传递的需要的数值,但是我的代码传成了tuple。学会的第一点,是堆栈的调用,我用的是pycharm,在代码下方就会显示调用的堆栈。最开始我根据之前对于整个流程的理解,以为这个调用就是从最后一个报错那里开始的,但实际上和我理解的正相反。是从main函数中调用test函数,然后一个一个找到的最后的。
在这个过程中,我们先理解了一下模型到底是怎么回事,后来发现,这个模型只有一个卷积层,然后经过计算,发现输入和输出的结果是一样的。但实际上在模型中有一步升采样没找到。之后通过理解了之后,正确的辨认return值和参数,找到了应该在那个步骤中,传递的值,第一个修改思路就是把return的六个值,改成一个,传递回去,但是这样在train的过程中就直接出错了,因为train是需要这六个参数的前两个的,所以得改回来,之后继续看传递的六个参数,和我们想在test传递的数不一样。因此设置了断点,去找到底应该是改成什么样的值,通过找lr_x2,lr_x4,的shape就发现了应该正确传递的值,因此最后改成了取y1,剩下的用默认值。
2019.10.8 Rcan代码修改心得
最新推荐文章于 2023-01-03 20:50:35 发布