目录
目录README.md

Jittor挑战热身赛CGAN

简介

本赛道将在数字图片数据集 MNIST 上训练 Conditional GAN(Conditional generative adversarial nets)模型,通过输入一个随机向量 z 和额外的辅助信息 y (如类别标签),生成特定数字的图像。

安装

运行环境

  • windows10
  • python>=3.7
  • jittor>=1.3.0

安装依赖

执行以下命令来安装python环境

pip install -r requirements.txt

训练

在当前目录下运行如下命令

python CGAN.py
关于

A Jittor implementation of Conditional GAN (CGAN).

29.0 KB
邀请码