Jittor

简介

使用MNIST数据集，训练 Conditional GAN，通过输入一个随机向量 z 和额外的辅助信息 y (如类别标签)，生成特定数字的图像。

环境

ubuntu 20.04 LTS python >= 3.7 jittor >= 1.3.0

部署

修改CGAN.py文件，将需要生成的数字赋予number变量 执行python CGAN.py