| Jittor第二届计图人工智能挑战赛 热身赛方案

Jittor第二届计图人工智能挑战赛 热身赛方案

|

简介

本项目包含了第二届计图挑战赛计图-热身赛的代码实现。主要方法基于CGAN，采用jittor方法对MNIST数据集处理，取得了生成电话号码的效果。

安装

本项目可在 1张GTX1660上运行，训练时间约为 30分钟。

运行环境

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

安装依赖

执行以下命令安装jittor ‘’’

pip install jittor

‘’’

预训练模型

预处理模型下载地址为https://pan.baidu.com/s/1u_QC21PcFbk-1akEmSoC7Q ，密码是1234，下载后放在目录/weights/下。

训练

模型的训练可以运行以下命令： python CGAN.py 训练完成后会根据设置的手机号自动进行推理。