目录
目录README.md

Jittor CGAN数字生成热身赛

主要结果

简介

本项目包含了第五届计图挑战赛热身赛比赛的代码实现。本项目的特点是:采用了在MNIST数据集上利用CGAN进行手写数字生成。

安装

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

运行环境

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

安装依赖

执行以下命令安装 python 依赖

sudo apt install python3.7-dev libomp-dev
python3.7 -m pip install jittor
# or install from github(latest version)
# python3.7 -m pip install git+https://github.com/Jittor/jittor.git
python3.7 -m jittor.test.test_example
pip install numpy

训练

| 介绍模型训练的方法

训练可运行以下命令:

python CGAN.py

推理

生成测试集上的结果可以运行以下命令:

python CGAN.py

致谢

此项目参考第五届计图人工智能挑战赛所给的baseline进行实现

注意事项

点击项目的“设置”,在Description一栏中添加项目描述,需要包含“jittor”字样。同时在Topics中需要添加jittor。

image-20220419164035639

关于

使用jittor框架进行手写数字生成

43.0 KB
邀请码