cliffwalker_CGAN_jittor

代码介绍

基于数据集 MNIST，训练一个将随机噪声和类别标签映射为数字图片的Conditional GAN模型，并生成指定数字序列。

仓库结构

• CGAN_jittor
  • images 存储训练过程的中间图片
  • saved_model 保存上一轮训练的模型
  • CGAN.py 模型训练&生成数字序列代码
  • result.png 生成的数字序列图片

    如何运行

    python -m pip install jittor
    python -m jittor_utils.install_cuda
    python -m jittor.test.test_core
    python -m jittor.test.test_example
    python -m jittor.test.test_cudnn_op
    python CGAN.py