CGAN_jittor

一个基于jittor框架的CGAN模型，使用MNIST数据集，可用于生成手写数字图片

运行方式

确保python版本在3.8以上

安装numpy和jittor库

pip install numpy
pip install jittor

运行程序

cd CGAN_jittor
python CGAN.py

可选参数

--n_epochs          number of epochs
--batch_size         size of each batch
--lr           learning rate
--b1           decay of first order momentum of gradient
--b2           decay of first order momentum of gradient
--n_cpu         number of cpu threads to use during batch generation
--latent_dim      dimensionality of the latent space
--n_classes      number of classes for dataset
--img_size       size of each image dimension
--channels       number of image channels
--sample_interval  interval between image sampling

训练结果

第n轮的训练结果保存在n000.png中，最终的训练结果保存在result.png中