CGAN-jittor-PPETVER

基于Jittor框架和MNIST数据集的CGAN实现

应用于第二届计图人工智能挑战赛热身赛

成绩：0.9996

Installation

安装依赖环境

pip install -r requirements.txt

Dependencies

• python
• Jittor

Usage

• Train

  • python CGAN.py

  • 运行时会输出当前batch的运行进度（如下图），每一个epoch后保存当前的运行效果图，每10个epoch会存储模型参数

  • 

• Test

  • 将Line137-193的训练部分代码（如下）注释

    for epoch in range(opt.n_epochs):
      print("epoch:", epoch)
      #...
      if epoch % 10 == 0:
      generator.save("generator_last.pkl")
      discriminator.save("discriminator_last.pkl")

    并运行下行指令

  • python CGAN.py

Resources

使用了比赛方提供的[框架代码](https://github.com/Jittor/gan-jittor/blob/master/ competition)，本仓库只进行了部分填充修改