使用Ray创建高效的深度学习数据管道-人工智能

文章编号:5795 技术教程 2023-11-03 深度学习Ray

扫码关注官方订阅号

使用Ray创建高效的深度学习数据管道人工智

首先,我们来看一下模型训练的伪代码

为了解决这个问题,可以将数据管道视为生产者——消费者的问题。数据管道生成小批量数据并写入有界缓冲区。模型/GPU从缓冲区中消费小批量数据,执行前向/反向计算并更新模型权重。如果数据管道能够以模型/GPU消费的速度快速生成小批量数据,那么训练过程将会非常高效。

图片

TensorFlowtf.dataAPI提供了一组丰富的功能,可用于高效创建数据管道,使用后台线程获取小批量数据,使模型无需等待。仅仅预先获取数据还不够,如果生成小批量数据的速度比GPU消费数据的速度慢,那么就需要使用并行化来加快数据的读取和转换。为此,Tensorflow提供了交错功能以利用多个线程并行读取数据,以及并行映射功能使用多个线程对小批量数据进行转换。

由于这些API是基于多线程的,所以可能会受到Python全局解释器锁(GIL)的限制。Python的GIL限制了一次只能运行单个线程的字节码。如果在管道中使用纯TensorFlow代码,通常不会受到这种限制,因为TensorFlow核心执行引擎在GIL的范围之外工作。但是,如果使用的第三方库没有解除GIL限制或者使用Python进行大量计算,那么依赖多线程来并行化管道就不可行

为了加快数据加载速度,可以使用多进程生成器。

如果进行分析,可以发现相当多的时间都花在了准备数据的反序列化上。在多进程生成器中,生产者进程会返回大型NumPy数组,这些数组需要进行准备,然后在主进程中进行反序列化。那么在进程间传递大型数组时,如何提高效率呢?

这就是Ray发挥作用的地方。Ray是一个用于在Python中运行分布式计算的框架。它带有一个共享内存对象存储区,可在不同进程间高效地传输对象。特别的是,在不进行任何序列化和反序列化的情况下,对象存储区中的Numpy数组可在同一节点上的worker之间共享。Ray还可以轻松实现数据加载在多台机器上的扩展,并使用ApacheArrow高效地序列化和反序列化大型数组。

Ray带有一个实用函数from_iterators,可以创建并行迭代器,开发者可以用它包装data_generator生成器函数。

以上就是使用Ray创建高效的深度学习数据管道的详细内容,更多请关注php中文网其它相关文章!

微信扫码关注PHP中文网服务号

QQ扫码加入技术交流群

Copyright2014-2023AllRightsReserved|苏州跃动光标网络科技有限公司|

全局中部横幅
草料二维码生成器

草料二维码把二维码技术变成简单实用的产品,让每个人都可以快速复用成功案例经验,自由组合内容展示、表单、批量、数据统计、美化和标签排版等功能,免费制作出能高效解决业务问题的二维码

6v电影

6v电影(www.hao6v.tv),每天搜集互联网最新电影和电视剧,为使用迅雷软件的用户提供最新的电影、电视剧、高清电影、免费下载等服务。

舜峰光电

舜峰光电是一家致力于光纤通信产品的设计、制造和销售的公司,公司主要生产产品包括:光纤阵列、光纤跳线、光纤分路器、光纤连接器等器件。与时俱进,创造品牌!面向世界,锻造辉煌!

雾华天气站

雾华天气站洞悉近期天气变化,提供详细的预报和分析,帮助用户在不同天气条件下做好应对准备。

塑料机械网

塑料机械网是国内塑料机械、橡胶机械领域专业B2B电子商务平台,提供注塑机,吹膜机,发泡设备,塑料辅助机械,塑机配件等行业资讯,产品价格,厂家新闻及技术资料等信息

瑞欧昂门窗

黑龙江省瑞欧昂科技有限公司,坐落于“百湖油城”大庆市萨尔图区,我公司临近大庆火车站、世纪大道,交通十分便捷,具有得天独厚的交通区位优势。我公司拥有多名经验丰富的中、高级技术人员,具有产品的研发、设计、生产、服务等综合能力。我公司是一家集加工、制作、销售、一体化多功能的厂家。

八图

八图H5,在线制作微信场景网站,h5在线制作工具,电子邀请函、婚礼邀请函、电子请柬、婚礼请柬、同学聚会、商务邀请函、会议邀请函、生日请柬、乔迁邀请函等海量H5一键生成网站轻松制作,不懂技术也可以制作酷炫微信h5页面

兼职网

兼职企鹅全国免费兼职招聘平台,您可以免费查看和发布网上兼职,周末兼职,寒暑期等兼职信息。兼职企鹅,全国免费兼职招聘平台。

天津朗津商贸有限公司

我们公司主要的特色服务是:汽车电瓶,汽车配件,汽车音响等,诚信是我们立足之本,创新是我们生存之源,便捷是我们努力的方向,用户的满意是我们大的收益、用户的信赖是我们大的成果。乘风破浪会有时,直挂云帆济沧海,耐驰人相信,面对新科技革命的激烈竞争,一定会创造出更加优秀的成绩,回报家乡、回报社会、回报祖国。

卸料除尘装置

本公司主要从事电液推杆、液压站、全自动液压纠偏装置、气动液压机械、各类闸门、物料防卡型三通、四通、钢包全程加揭盖系统以及各种非标产品的设计、生产与销售。

全局底部横幅