Jittor 计图挑战热身赛(手写数字生成)

简介

本项目包含了第四届计图挑战赛热身赛的代码实现。本项目的特点是：使用 Jittor 库训练生成对抗网络（GAN）模型，针对 MNIST 数据集进行训练。

安装

本项目可在 1 张 RTX 3060 上运行，训练时间约为1小时。

运行环境

• windows 10
• python >= 3.7
• jittor >= 1.3.0

安装依赖

执行以下命令安装 python 依赖

python -m pip install jittor

训练

修改number变量值

python GGAN.py

推理

使用 Jittor 训练 GAN 模型在 MNIST 数据集上生成逼真图像，使用不同号码生成图片进行测试

致谢

根据计图官方代码填充