PA3
This is a project of generating images of handwritten digits using Conditional Generative Adversarial Networks (CGAN) and the MNIST dataset.
Use the package manager pip to install jittor.
sudo apt install python3.7-dev libomp-dev python3.7 -m pip install jittor python3.7 -m jittor.test.test_example # 如果您电脑包含Nvidia显卡，检查cudnn加速库 python3.7 -m jittor.test.test_cudnn_op
python3.7 CGAN.py
You can use the following options to modify the parameters of the model:
python3.7 CGAN.py --n_epochs 250 --batch_size 128 --lr 0002 --b1 0.5 --b2 0.999 --n_cpu 8 --latent_dim 100 --n_classes 10 --img_size 32 --channels 1 --sample_interval 1000
Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.
Please make sure to update tests as appropriate.
MIT
PA3_CGAN
This is a project of generating images of handwritten digits using Conditional Generative Adversarial Networks (CGAN) and the MNIST dataset.
Installation
Use the package manager pip to install jittor.
Usage
You can use the following options to modify the parameters of the model:
Results
Contributing
Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.
Please make sure to update tests as appropriate.
License
MIT