Jittor 计图挑战热身赛

简介

本项目包含了第三届计图挑战赛计图 - 计图挑战热身赛的代码实现。本项目在数字图片数据集MNIST上训练Conditional GAN模型，通过输入一个随机向量 z 和额外的辅助信息 y (如类别标签)，生成特定数字的图像。

安装

本项目可在 1 张 1080ti 上运行，训练时间约为 20 分钟。

运行环境

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

安装依赖

执行以下命令安装 jittor 依赖

sudo apt install python3.7-dev libomp-dev
python3.7 -m pip install jittor
python3.7 -m jittor.test.test_example
# 如果您电脑包含Nvidia显卡，检查cudnn加速库
python3.7 -m jittor.test.test_cudnn_op

训练

python CGAN.py

致谢

本项目基于计图（jittor）官方示例代码完成