Jittor挑战热身赛 baseline
简介
本项目包含了第三届计图挑战热身赛的代码实现,采用了 CGAN 方法对 MNIST数据集处理,使其生成自己想要生成的数字。
安装
- python >= 3.7
- jittor >= 1.3.0
训练与推理
运行CGAN-jittor.py文件即可进入模型训练、推理的过程
在此之前,需要更改上述代码第173行generator.load、174行discriminator.load的路径,以及190行最终推理结果存储的路径
致谢
此项目部分代码参考了计图挑战热身赛提供的示例代码。
Jittor挑战热身赛 baseline
简介
本项目包含了第三届计图挑战热身赛的代码实现,采用了 CGAN 方法对 MNIST数据集处理,使其生成自己想要生成的数字。
安装
训练与推理
运行CGAN-jittor.py文件即可进入模型训练、推理的过程 在此之前,需要更改上述代码第173行generator.load、174行discriminator.load的路径,以及190行最终推理结果存储的路径
致谢
此项目部分代码参考了计图挑战热身赛提供的示例代码。