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日)

在这里插入图片描述

相关文章
|
弹性计算 IDE 安全
打造在线编程环境
打造自己的在线编程环境,使用Coding WebIDE来实现在线的IDE,以便我们在不方便使用本地编辑器的时候可以使用在线编辑器来进行我们的在线编辑操作。
打造在线编程环境
|
3月前
|
Java 编译器 Maven
【颠覆你的认知!】当Quarkus邂逅GraalVM本机镜像,应用启动竟快到飞起——背后的技术秘密等你揭秘!
Quarkus框架因轻量级与高性能而在Java开发社区广受关注。结合GraalVM使用能显著提升应用启动速度与运行效率,这得益于GraalVM的本机镜像支持。本文将介绍如何利用Quarkus和GraalVM构建高效应用,并提供示例代码演示具体步骤。首先需安装GraalVM环境并配置Maven支持构建本机镜像。接着创建一个简单的RESTful服务端点作为示例,通过命令行编译生成本机可执行文件并运行。这种方式能够大幅提升应用性能,但需注意构建时间和部分Java特性兼容性问题。
54 1
|
消息中间件 架构师 算法
好家伙!全网开源对标P5~P7程序员技术成长路线,阿里这次是真卷
今年的程序员可以说是最焦虑的一个群体了,面试找工作投简历没人理,有面试机会也面试不过,面试进去还干不长...于是,程序员们纷纷直呼:互联网寒冬又双叒叕来了,环境不好努力也没用躺平算了。
283 1
好家伙!全网开源对标P5~P7程序员技术成长路线,阿里这次是真卷
|
设计模式 Cloud Native Java
GitHub爆出阿里内部突击路线+配套学习资源,直接霸榜
又到了一年一度的备战秋招的时间,虽然这两年因为经济环境不太好,年初互联网行业的各大厂都再裁员,前半年的行业发展趋于平稳,随着经济形势正在复苏,今年秋招各个公司hc也逐渐开放,很多小伙伴也反馈面试机会变多了,想查缺补漏冲刺大厂,我软磨硬泡终于从阿里p8手里弄来了突击秘籍,分享给大家。 涨薪秘籍从Java核心并发开始,到经典的框架源码设计模式、分布式微服务架构设计、微服务NetFilx体系和Alibaba体系、微服务架构性能调优、云原生以及虚拟化、Spring Cloud Alibaba项目实战。
171 2
GitHub爆出阿里内部突击路线+配套学习资源,直接霸榜
|
人工智能 并行计算 架构师
CUDA编程模型都改了!英伟达架构师团队撰文详解:Hopper为啥这么牛?
CUDA编程模型都改了!英伟达架构师团队撰文详解:Hopper为啥这么牛?
369 0
|
前端开发 Android开发
我奇怪的安卓开发历程(内含本人不太成熟的技术观)(四)
工作一天下来,最大的收获便是我发现项目写了半年我还是不知道这是个啥架构的,通过几次和同事沟通交流,以及浏览许多介绍和实例MVP、MVVM架构的文章,还是不了了之,当然,不能否认的是同事认为这是MVP架构的项目,我也挺倾向这个观点。
|
程序员 开发工具 Android开发
我奇怪的安卓开发历程(内含本人不太成熟的技术观)(三)
半年的开发工作通过统计git的提交记录发现其实我提交行数最多的项目也就1228行:
108 0
|
设计模式 缓存 算法
阿里逆天级调优方案,内部这套Java性能调优实战宝典,堪称教科书
随着互联网的发展,高可靠、高并发以及降本增效,已成为各大公司面临的现实挑战,性能优化需求愈发迫切,大到分布式系统,小到代码块的算法优化,都已经成为你日常工作中必须要面对的事情。对于开发者而言,性能优化也从加分项变为一个热门技能,缺乏相关知识将很难在面试或工作中脱颖而出
阿里逆天级调优方案,内部这套Java性能调优实战宝典,堪称教科书
|
存储 Android开发
我奇怪的安卓开发历程(内含本人不太成熟的技术观)(二)
初出茅庐的我在开发工作中成长,收获了许多,但其实也引出了一堆问题。
111 0

热门文章

最新文章