TransGAN代码复现—九天毕昇平台

简介: TransGAN是UT-Austin、加州大学、 IBM研究院的华人博士生构建了一个只使用纯 transformer 架构、完全没有卷积的 GAN,并将其命名为 TransGAN。论文地址:https://arxiv.org/abs/2102.07074论文精读:论文精读 TransGAN代码地址:https://github.com/VITA-Group/TransGAN本篇博文是利用中国移动的九天毕昇平台复现TransGAN的博文。

@TOC

TransGAN是UT-Austin、加州大学、 IBM研究院的华人博士生构建了一个只使用纯 transformer 架构、完全没有卷积的 GAN,并将其命名为 TransGAN。

论文地址:https://arxiv.org/abs/2102.07074

论文精读:论文精读 TransGAN

代码地址:https://github.com/VITA-Group/TransGAN

本篇博文是利用中国移动的九天毕昇平台复现TransGAN的博文。

一、九天毕昇操作简介

首先介绍一下深度学习服务平台:“九天·毕昇”是中国移动开发的一款云服务平台,能够提供高效(最高显存32GB的V100)便捷(主要优点还是免费O(∩_∩)O)的算力。

点击链接注册即可免费赢取 1000 个算力豆(50 小时 V100 使用时长: https://jiutian.10086.cn/web#/register?token=NTIxNWNkOWEtNjIxZi00ZTIyLThlZTItNDhiYWQ3ODRiMDhm

1.1、进入控制台

进入首页后,点击进入控制台:

在这里插入图片描述

在这里插入图片描述

进入后可以看到存储空间,算力豆(九天毕昇算时的计算单位)和自己建立过的实例。

1.2、新建自己的实例

在这里插入图片描述

点击左侧导航栏,进入模型训练,点击新建实例

在这里插入图片描述

命名自己的实例名称,然后选择数据集(平台提供了一些数据集,选用平台数据集会自动加载在data/public文件夹中,但是平台只有

部分数据集,没有的数据集需要自己在数据管理或者在运行实例时进行上传)选择资源套餐。

1.3、进入实例

进入实例后,我们可以看到,和colab比较类似,左侧是文件资源部分,右侧可以进行编辑,

在这里插入图片描述

我们点击左上角的蓝色加号,新建一个notebook或者console或者最下面的terminal(看个人喜好),如果点击的是pytorch1.x,则平台就自动为了加载了最基础的pytorch环境。

在这里插入图片描述

二、复现TransGAN

2.1、下载git代码

首先进入终端terminal,克隆代码:

git clone https://github.com/VITA-Group/TransGAN

在这里插入图片描述

成功克隆后,项目代码如下:

在这里插入图片描述

2.2、配置环境

进入TransGAN:cd TransGAN,

然后根据requirement.txt下载安装环境:

pip install -r requirements.txt

可能会出现的问题:

ERROR: conda 4.11.0 requires ruamel_yaml_conda>=0.11.14, which is not installed.

ERROR: tensorflow 2.4.0 has requirement absl-py~=0.10, but you'll have absl-py 1.0.0 which is incompatible.

ERROR: tensorflow 2.4.0 has requirement flatbuffers~=1.12.0, but you'll have flatbuffers 2.0 which is incompatible.

ERROR: tensorflow 2.4.0 has requirement gast==0.3.3, but you'll have gast 0.5.3 which is incompatible.

ERROR: tensorflow 2.4.0 has requirement grpcio~=1.32.0, but you'll have grpcio 1.45.0 which is incompatible.

ERROR: tensorflow 2.4.0 has requirement h5py~=2.10.0, but you'll have h5py 3.6.0 which is incompatible.

ERROR: tensorflow 2.4.0 has requirement numpy~=1.19.2, but you'll have numpy 1.21.5 which is incompatible.

ERROR: tensorflow 2.4.0 has requirement six~=1.15.0, but you'll have six 1.16.0 which is incompatible.

ERROR: tensorflow 2.4.0 has requirement typing-extensions~=3.7.4, but you'll have typing-extensions 4.1.1 which is incompatible.

解决方案:这个的意思是tensorflow 2.4版本和一些其他环境的版本不能匹配。

需要重新安装TensorFlow:

pip install tensorflow

2.3、开始训练

在TransGAN目录下,输入:

python exps/cifar_train.py

开始训练:

在这里插入图片描述

训练的时间比较长,可能需要2天或者更久,我们也可以下载已经训练好的checkpoint进行测试(请看2.4)。

2.4、开始测试

如果2.3没有训练或者完,下载 cifar checkpoint并将其放到文件夹中:

在这里插入图片描述

最后输入测试命令,开始测试:

python exp/cifar_test.py

可能遇见的问题:

ModuleNotFoundError: No module named 'tensorflow._api'

解决方案1:

参考网上推测要在命令行装tensorflow-gpu,在命令行输入:

pip install --upgrade --ignore-installed tensorflow-gpu

解决方案2:

TensorFlow未安装成功,重新pip安装

福利

点赞、评论+收藏三连,可以私信我领取专属算力豆!(活动截止2022年6月29日)

在这里插入图片描述

目录
打赏
0
0
0
0
12
分享
相关文章
算子开发到推理加速,一位00后开发者的“升级打怪”之旅
郑辉,从大四学生到资深AI开发者,他的成长轨迹映射了中国AI生态的繁荣。他独立开发的NonZero算子被纳入CANN主线,参与的深度学习框架降低了大模型训练门槛。在昇腾AI生态中,郑辉们正助力大模型发展。从初次接触分布式机器学习到优化算子,再到参与大模型推理加速项目,郑辉在实践中不断成长,他的工作有望帮助开发者更高效地部署在线推理服务。他的故事展示了开发者如何通过团队合作和持续学习,克服挑战,推动AI在各行业的应用,照亮智能化之路。
157 0
惊爆!Uno Platform 调试与性能分析终极攻略,从工具运用到代码优化,带你攻克开发难题成就完美应用
【8月更文挑战第31天】在 Uno Platform 中,调试可通过 Visual Studio 设置断点和逐步执行代码实现,同时浏览器开发者工具有助于 Web 版本调试。性能分析则利用 Visual Studio 的性能分析器检查 CPU 和内存使用情况,还可通过记录时间戳进行简单分析。优化性能涉及代码逻辑优化、资源管理和用户界面简化,综合利用平台提供的工具和技术,确保应用高效稳定运行。
149 0
Bun 1.0 正式发布,爆火的前端运行时,速度遥遥领先!
Bun 1.0 正式发布,爆火的前端运行时,速度遥遥领先!
472 0
打跑个人开源转角的“拦路虎”——我的开源之路
打跑个人开源转角的“拦路虎”——我的开源之路
84 0
极智AI | 全志XR806 OpenHarmony鸿蒙系统固件编译
大家好,我是极智视界,本教程详细记录了全志 XR806 OpenHarmony 鸿蒙系统固件编译的方法。
153 0
CUDA编程模型都改了!英伟达架构师团队撰文详解:Hopper为啥这么牛?
CUDA编程模型都改了!英伟达架构师团队撰文详解:Hopper为啥这么牛?
414 0
本地没有环境跑深度学习模型? 阿里云天池实验室它不香吗
前几天做深度学习模型训练使用 Google 的 colab 总是掉,搞得很烦。然后那天我队友 “叶伏天” 和我说有一个类似于 Google colab 的平台,可以训练,虽然也有 8 小时的限时,但是这两个我可以替换使用,甚至一起跑实验,显卡配置也可以满足我的需求训练模型的需求
3207 0
本地没有环境跑深度学习模型? 阿里云天池实验室它不香吗
真香!Github一夜爆火,阿里性能优化不传之秘终于开源
性能调优,是从开发岗跃迁至架构岗的拦路虎。如果你是一个小白,那么恭喜你性能优化这个东西你暂时还不需要扛着。但如果你是公司的中坚力量,想把技术水平往架构层面靠近,那么性能优化这个东西你必须要扛下来,并且解决它。
声称掌握核心技术,实则换皮Python?国产编程语言木兰引起社区热议
近日有媒体称,一款面向智能物联网和编程教育的国产编程语言出世,名为木兰。消息一出,引起了开发者社区的讨论。但有人发现,下载的程序包解包后——是个 Python?
501 0
声称掌握核心技术,实则换皮Python?国产编程语言木兰引起社区热议