Jittor 计图挑战热身赛 baseline

简介

本项目包含了第三届计图挑战赛计图 - 热身赛的代码实现。本项目的特点是：采用了 Conditional GAN(Conditional generative adversarial nets) 方法对 MNIST数据集进行训练处理，并生成特定数字的图像。

安装

参考jittor框架。

本项目可在 1 张 3080 上运行,训练时间约为 13分钟。

运行环境

• ubuntu 20.04.6 LTS
• Python 3.8.10
• jittor

运行

｜ 介绍模型训练的方法

运行以下命令：

Python CGAN.py

致谢

此项目的部分代码参考了 jittor-jgan。