jittor_budui_warmup

简介

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

安装

运行环境

• 操作系统: Ubuntu >= 16.04 或 Windows Subsystem of Linux（WSL）
• Python：版本 >= 3.7
• C++编译器 （需要下列至少一个）
• g++ （>=5.4.0）
• clang （>=8.0）
• GPU 编译器（可选）：nvcc >=10.0
• GPU 加速库（可选）：cudnn-dev

训练

可运行以下命令：

python CGAN.py