jittor

jittor 手写数字生成baseline 项目名称 jittor-【我该怎么应对】-【jittor热身赛】 主要结果：请看仓库里的图片

简介 该项目为第四届计图人工智能挑战赛-热身赛的代码实例–利用cgan模型生成特定数字的图片 本项目的特点是：优化了训练训练层数以及平滑度等各个参数进行模型训练。

安装 本项目使用CPU|(amd r5 6600h)运行，运行时间在5-6个小时之间

运行环境 ubuntu 20.04 LTS python >= 3.7 jittor >= 1.3.0

安装依赖 执行以下命令安装 python 依赖 pip install -r requirements.txt

数据处理：无，使用了mnist数据集进行模型的训练

训练 单卡训练可运行以下命令： python CGAN.py

推理 可通过调用如下代码生成指定数字字符的手写数字照片 number = “要生成的数字序列”

注意事项 如果你有英伟达显卡，还是用显卡训练好点。

许可证 Apache License2.0