jittor_warm_up

本项目是第二届计图挑战赛的热身赛项目，项目使用赛事官方给定的jittor框架下的Conditional GAN，通过在MNIST数据集上训练，可生成指定的手写数字图像。

安装

运行环境

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

安装依赖

执行以下命令安装 python 依赖

sudo python3.7 -m pip install -r requirements.txt

训练&运行

使用 修改CGAN.py中的号码

python CGAN.py

致谢

代码基于赛事官方代码完成 https://github.com/Jittor/gan-jittor/blob/master/models/cgan/cgan.py