cycleGAN生成图片问题

来源:9-7 cycleGAN模型搭建-test

慕2003029

2020-11-17

老师您好,我在训练cycleGAN时遇到了些问题。当我只训练1个epoch时测试,可以得到如下图结果:图片描述图片描述
效果虽然不完美,但接近理想。当训练批次增加到100个epoch时,测试生成的结果却如下图:图片描述图片描述
然后我继续训练,直到epoch达到200也没有改善,请问老师出现这种问题是什么原因呢?

写回答

1回答

会写代码的好厨师

2020-11-26

可以提供下loss曲线。看样子训练出现了崩塌的现象。

0
1
慕2003029
老师您好,我将您train.py中的源代码第123行和第137行的loss_D_fake = loss_GAN(pred_real, label_fake)改为loss_D_fake = loss_GAN(pred_fake, label_fake)解决了问题,可以训练出很理想的结果
2020-12-01
共1条回复

PyTorch入门到进阶 实战计算机视觉与自然语言处理项目

理论基础+技术讲解+实战开发,快速掌握PyTorch框架

1190 学习 · 293 问题

查看课程