Jittor 计图挑战热身赛

简介

使用MNIST数据集训练CGAN模型，生成规定数字ID的图片

运行环境

• Windows 10 professional
• Python 3.9.15
• CUDA 11.8

安装依赖

pip install jittor
conda install pywin32(if use conda env)

运行

修改CGAN.py中的number变量，之后执行python CGAN.py

参考

基于计图官方框架填充完成