RCAN, EDSR代码dataloader.py出现错误

环境：python3.7+ cuda10.0 + pytorch1.1.0

在运行MSRN代码时出现老多问题了，解决的路上全是心酸泪我的妈呀！

在此记录下来，希望朋友们少走点弯路

1. 问题：ImportError: cannot import name ‘_update_worker_pids’

   原因：MSRN中的dataloader文件中会出现很多这类问题，找不到对应的函数名和属性名等等，因为Pytorch升级到1.0后，dataloader模块中的函数有修改

   解决：参考并非常感谢这位博主以及这位博主

2. 问题：THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=385 error=11 : invalid argument

   解决：a. 删除之前的用conda创建的pytorch相关环境

            b. 用pytorch官网的pip3命令重新安装pytorch和torchvision
   
            c. 另外重新安装所有相关的包
   

3. 问题：ValueError: Cannot load file containg pickled data when allow_pickle=False

   解决：参考并非常感谢这位博主

4. 问题：AttributeError: ‘SimpleQueue’ object has no attribute cancel_join_thread()

   原因：源代码中调用了multiprocessing.SimpleQueue()得到了worker_result_queue变量，之后这个变量在pytorch自带的dataloader中需要去调用cancel_join_thread（）方法，但这个方法其实是multiprocessing中Queue类的方法，SimpleQueue并没有！难道是pytorch更新版本后没有的？？？

   解决：将Train文件和Test文件夹下的dataloader.py中的multiprocessing.SimpleQueue()都改为multiprocessing.Queue()即可，至于原来的SimpleQueue有多好用我就不管了。。

5. 问题：ValueError:x and y must have same first dimension, but have shapes(2,) and (1,)

   原因： 找了半天原因，突然想起训练的时候为啥显示epoch从2开始，感觉不对劲，于是

   解决：将Train/trainer.py中三处 self.scheduler.last_epoch + 1 改为 self.scheduler.last_epoch

6. 问题：第lr_decay倍的epoch的lr是上一个的0.25倍，但应该是0.5倍来着

   原因：在self.scheduler.step(）时就已经计算了新epoch的lr，然而之后在lr = self.scheduler.get_lr()[0]时又计算了一遍。

   解决：lr = self.scheduler.get_lr()[0] 修改为

lr = self.optimizer.param_groups[0][‘lr’]

7. 问题：其实接着6还有问题。。按照6修改之后，不是每lr_decay个epoch修改一次lr，而是1～lr_decay-1为相同的lr，lr_decay～2lr_decay-1为相同的0.5lr，依次类推，我觉得这样错一个也行，但还是有点手痒啊于是。。

   解决：

   self.scheduler.step()
   self.loss.step()
   epoch = self.scheduler.last_epoch

   修改为

   self.scheduler.step(self.scheduler.last_epoch)
   self.loss.step()
   self.scheduler.last_epoch += 1
   epoch = self.scheduler.last_epoch

8. 问题：接着7改完后又发现新的问题，如果训练中断在运行self.ckp.write_log(…）这句之后，继续训练时如果正好该下降lr了，那么

以上，终于测试和训练都成功运行！！！榨干了我的精华！
————————————————
版权声明：本文为CSDN博主「睡饱的大宝」的原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/sinat_31987445/article/details/91399956

ference
[2]: https://mermaidjs.github.io/
[3]: https://mermaidjs.github.io/
[4]: http://adrai.github.io/flowchart.js/