目录
目录README.md

ConditionalGAN

提供数字图片数据集 MNIST,训练一个将随机噪声和类别标签映射为数字图片的Conditional GAN模型,并生成指定数字序列。

提供示例代码框架,提供数据下载、模型定义、训练步骤等功能。

可以运行如下命令并加入任意参数进行训练

python CGAN.py 
python CGAN.py --n_epoches <n> --batch_size <b> 

可以用-h得到所有可选参数

python CGAN.py -h(--help)
关于

A Jittor implementation of Conditional GAN (CGAN).

29.0 KB
邀请码