Jittor 计图挑战热身赛 CGAN

简介

本赛题将会提供数字图片数据集 MNIST，参赛选手需要训练一个将随机噪声和类别标签映射为数字图片的Conditional GAN模型，并生成比赛页面指定数字序列。

本赛题提供示例代码框架，提供数据下载、模型定义、训练步骤等功能。

安装

运行环境

• Windows 11
• Python >= 3.8
• Jittor >= 1.3

安装依赖

pip install -r requirements.txt

训练与运行

python CGAN.py
使其运行