1.运行python scripts/script.py --predict时出现 cuda out of memory ERROR 错误如下图
解决方法:在script.py开头加上如下图代码
2.由于pytorch版本问题,将data[0]均换成item()
运行python scripts/script.py --train时出现错误:
将single_model.py中代码改为如下图
至此,小白我的enlightenGAN源码终于能跑起来了!接下来就是改动提出创新点了!虽说后面改动肯定还会出现很多错误,但能取得阶段性的成功还是非常开心的!
如果有需要代码和环境信息的可以踢我,可以分享给大家,只记录了部分复现遇到的bug。真心希望每个想复现代码的兄弟姐妹都能顺顺利利,远离bug,太痛苦了!!!