jittor 热身赛 CGAN

主要结果

简介

jittor热身赛，在数字图片数据集 MNIST 上训练 Conditional GAN(Conditional generative adversarial nets)模型，通过输入一个随机向量 z 和额外的辅助信息 y(如类别标签)，生成特定数字的图像。

安装

运行环境

Windows
Python>=3.7
jittor>=1.3.0

安装依赖

pip install jittor

训练

运行CGAN.py文件即可开始训练，训练结果会生成手机号码（可自行修改代码生成目标手机号码）。