CGAN

使用 Jittor 实现 Conditional GAN (CGAN).

Requirements

运行代码需要安装 Jittor 框架.

Running

python CGAN.py

可选参数及含义：

--n_epochs (number of epochs of training)
--batch_size (size of the batches)
--lr (adam: learning rate)
--b1 (adam: decay of first order momentum of gradient)
--b2 (adam: 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)