基于jittor框架的CGAN实现

环境要求

python 版本大于3.8

安装对应版本的jittor网址：https://cg.cs.tsinghua.edu.cn/jittor/download/

如何运行

python CGAN.py即可运行

options

options:
  -h, --help            show this help message and exit
  --n_epochs N_EPOCHS   number of epochs of training
  --batch_size BATCH_SIZE
                        size of the batches
  --lr LR               adam: learning rate
  --b1 B1               adam: decay of first order momentum of gradient
  --b2 B2               adam: decay of first order momentum of gradient
  --n_cpu N_CPU         number of cpu threads to use during batch generation
  --latent_dim LATENT_DIM
                        dimensionality of the latent space
  --n_classes N_CLASSES
                        number of classes for dataset
  --img_size IMG_SIZE   size of each image dimension
  --channels CHANNELS   number of image channels
  --sample_interval SAMPLE_INTERVAL
                        interval between image sampling