第三届计图人工智能挑战赛

计图挑战热身赛

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

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

安装

| 介绍基本的硬件需求、运行环境、依赖安装方法

本项目在 单张3090 上运行，训练时间约为 半小时。

运行环境

• ubuntu 18.04
• python >= 3.8
• jittor >= 1.3.1

选手可以基于示例代码填充注释为 TODO 的部分完成该赛题。

修改 CGAN.py 
python CGAN.py
使其运行