A Jittor Impl of Conditional-GAN, based on MNIST dataset.

Result

Intro

This is a simple implementation of Conditional-GAN, based on MNIST dataset. The generator is a simple MLP, and the discriminator is a simple CNN.

Setting up development env

• Install Jittor framework (idk how to)
• Check jittor cuda support

Run

python CGAN.py

Code adapted from

Jittor official examples