Jittor 热身赛 CGAN

简介

本项目包含了第二届计图挑战赛计图 - 热身赛的代码实现。

安装

本项目可在 CPU 上运行，训练时间约为 20分钟。

运行环境

• ubuntu 20.04 LTS
• python >= 3.7
• jittor >= 1.3.0

安装依赖

执行以下命令安装 python 依赖

pip install -r requirements.txt

预训练模型

预训练模型生成器为generator_last.pkl判别器为discriminator_last.pkl

训练&推理

生成测试集上的结果可以运行以下命令：

python CGAN.py