目录
目录README.md

jittor-ST605-warmup

| 第二届计图挑战赛-热身赛开源

Jittor 热身赛-手写体数字生成 baseline

简介

本项目包含了第二届计图挑战赛计图 - 热身赛(手写数字生成)比赛的代码实现。

安装

| 介绍基本的硬件需求、运行环境、依赖安装方法

本项目在 1 张 3070 上运行,训练时间约为 20 分钟。

运行环境

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

安装依赖

执行以下命令安装 python 依赖

pip install -r requirements.txt

预训练模型

预训练模型模型下载地址为 https:abc.def.gh,下载后放入目录 <root>/weights/ 下。

训练

单卡训练可运行以下命令:

python competition/warm_up_comp/CGAN.py

致谢

此项目基于论文 Conditional Generative Adversarial Networks 实现,部分代码参考了 jittor-gan

关于

jittor 生成手写体数字 ST605 队

33.0 KB
邀请码