今天想讲一讲CGAN项目实现过程中遇到的问题
经过写代码,改bug,配环境等一系列折磨,我终于可以开始训练我的代码啦哈哈哈哈哈
激动的心,颤抖的手
然而........
当程序开始运行,损失慢慢变小,看到它生成的图像越来越逼近真实图像,心里很高兴很激动。就在我觉得只需要等待就可以得到想要的结果时,生成器与判别器损失变为了NAN,一时间不知如何是好。结果就导致了我生成的图像效果并不好。从那一刻开始,我开启了我的寻找问题之旅。
经过多天的茫然,我貌似是找到了问题所在:梯度消失?模式崩溃?梯度爆炸?
总而言之,GAN模型很难训练,很容易造成以上问题。如何解决呢?
改进GAN网络,据说WGAN可以有效缓解以上问题,因而我开始学习WGAN啦,至于结果怎样,等我学完再来告诉大家~
附一张心酸小图片呜呜呜呜