Jittor挑战赛 - 热身赛

简介

本项目包含了第二届计图挑战赛计图 - 热身赛的代码实现。 本项目基于jittor深度学习框架，训练了一个CGAN模型，用以生成手写数字图片。

安装

本项目在CPU上运行，在 2 核 i5-8500 下的训练时间约为 2 小时。

运行环境

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

安装jittor框架

执行以下命令安装 jittor框架

sudo apt install python3.8-dev libomp-dev  
python3.8 -m pip install jittor  

数据预处理

无

训练

执行以下命令

python3.8 CGAN.py

推理

无

致谢

项目baseline由 jittor-gan提供。