Jittor 计图挑战热身赛 baseline

简介

本项目包含了第二届计图挑战赛计图 - 计图挑战热身赛的代码实现。本项目的特点是：采用了CGAN方法在MNIST数据集上训练，实现手写数字的条件式生成。

安装

本项目可在单张1080Ti上运行。

运行环境

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

安装依赖

执行以下命令安装python依赖：

pip install -r requirements.txt

训练

修改CGAN.py中的number为自己的手机号，训练可运行以下命令：

python CGAN.py

致谢

此项目基于论文 Conditional Generative Adversarial Nets 实现，部分代码参考了 jittor-gan。