当前位置:   article > 正文

GAN的loss为nan的解决办法(成功)_gan网络输入不平衡数据训练loss为nan

gan网络输入不平衡数据训练loss为nan

跑了好几个GAN模型,一运行loss就巨大,几次epoch后就变NaN,或者直接就爆掉
在这里插入图片描述
有人说loss为nan是因为里面有log出问题了,但我想着文章里给的代码肯定得是能跑的啊

后来换了个电脑,重新把cyclegan的源码跑了一下,又完全好了
震惊!
后来我对比了一下底层,发现能跑通的这个跟其他的连conv2d的底层都不一样,所以判断是package的版本不一样
于是我把所有其他GAN代码的环境都变成能跑通的这个
其他的GAN也就都能跑通了

声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号