2019.10.8 Rcan代码修改心得

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

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 11
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 11
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值