目录
目录README.md

Jittor CGAN

主要结果

使用Conditional GAN算法生成手写数字

简介

本项目包含了第五届计图挑战赛(热身赛)的代码实现。本项目的特点是:采用了Conditional GAN算法,在MNIST数据集上训练,实现了手写数字的生成。

安装

本项目需要在NVIDIA GPU上运行,且需要支持cuda。

运行环境

  • python >= 3.7
  • jittor >= 1.3.0

安装依赖

参照cg.cs.tsinghua.edu.cn/jittor/download/

训练

运行命令:python CGAN.py [arguments]

关于

使用 Conditional GAN 算法生成手写数字,通过 jittor 机器学习框架实现

37.0 KB
邀请码