Jittor 热身赛 - Conditional GAN

简介

本项目包含了第四届计图挑战赛计图热身赛。

在 MNIST 数据集上训练 Conditional GAN (Conditional Generative Adversarial Network) 模型，通过输入随机向量 z 和一个额外的辅助信息 y 生成特定数字的图像。

运行环境

• ubuntu 20.04 LTS
• python >= 3.7
• jittor >= 1.3.0

方法

终端运行 python CGAN.py 即可。