CGAN_jittor_shzr

题目目标

使用生成指定数字序列对应的图片。

方法

使用CGAN网络，生成器G用于生成图片，判别器D用于判别真实图片和生成图片。对于D，需要训练最大化loss，用于区分真实图片和生成图片。对于G，需要训练最小化loss，用于生成和真实图片相似的图片。

运行方法

python CGAN.py

由于连续训练100轮会导致内存占用过大崩溃，因此我每训练20轮后会保存当前模型，然后重新加载模型继续训练。

结果

最终训练得到的模型见 ./G_last.pkl, ./D_last.pkl