目录
目录README.md

Jittor挑战热身赛 baseline

简介

本项目包含了第三届计图挑战热身赛的代码实现,采用了 CGAN 方法对 MNIST数据集处理,使其生成自己想要生成的数字。

安装

  • python >= 3.7
  • jittor >= 1.3.0

训练与推理

运行CGAN-jittor.py文件即可进入模型训练、推理的过程 在此之前,需要更改上述代码第173行generator.load、174行discriminator.load的路径,以及190行最终推理结果存储的路径

致谢

此项目部分代码参考了计图挑战热身赛提供的示例代码。

关于

计图挑战热身赛代码开源

37.0 KB
邀请码