计图挑战热身赛icter

简介

本项目包含了第四届计图人工智能挑战赛热身赛icter团队的代码实现。

本项目的特点是：采用了 GAN 方法对用户随机ID 20190612014308进行生成，取得了如上生成效果。

安装

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

运行环境

• ubuntu 20.04 LTS
• python = 3.7.16
• jittor = 1.3.9.4

# 创建虚拟环境
conda create -n icter python=3.7
# 安装最新版本 jittor
git clone https://github.com/Jittor/jittor.git
sudo pip install ./jittor

如果安装jittor过程中遇到 Check failed ret(256) == 0(0)问题，请参考博客：正确安装 cuda, cudnn 和 安装 jittor，解决错误：Check failed ret(256) == 0(0) 进行解决。

训练

训练可运行以下命令：

python CGAN.py

致谢

此项目基于比赛主办方发布在 第四届计图人工智能挑战赛 官方网站发布的 示例代码 基础上完成。