关于Tensorflow分布式并行策略

作者:YongqiangGao 时间:2024-01-02 09:39:28 

tensorFlow中主要包括了三种不同的并行策略,其分别是数据并行、模型并行、模型计算流水线并行,具体参考Tenssorflow白皮书,在接下来分别简单介绍三种并行策略的原理。

数据并行

一个简单的加速训练的技术是并行地计算梯度,然后更新相应的参数。数据并行又可以根据其更新参数的方式分为同步数据并行和异步数据并行,同步的数据并行方式如图所示,tensorflow图有着很多的部分图模型计算副本,单一的客户端线程驱动整个训练图,来自不同的设备的数据需要进行同步更新。这种方式在实现时,主要的限制就是每一次更新都是同步的,其整体计算时间取决于性能最差的那个设备。

关于Tensorflow分布式并行策略

数据并行还有异步的实现方式,如图所示,与同步方式不同的是,在处理来自不同设备的数据更新时进行异步更新,不同设备之间互不影响,对于每一个图副本都有一个单独的客户端线程与其对应。在这样的实现方式下,即使有部分设备性能特别差甚至中途退出训练,对训练结果和训练效率都不会造成太大影响。但是由于设备间互不影响,所以在更新参数时可能其他设备已经更好的更新过了,所以会造成参数的抖动,但是整体的趋势是向着最好的结果进行的。所以说这种方式更适用于数据量大,更新次数多的情况。

关于Tensorflow分布式并行策略

模型并行

一个模型并行训练的例子如图所示,其针对的训练对象是同一批样本数据,但是将不同的模型计算部分分布在不同的计算设备上同时执行。

关于Tensorflow分布式并行策略

模型计算流水线并行

此并行方式主要针对在同一个设备中并发实现模型的计算,如图是其并发计算步骤,可以发现它实际上与异步数据并行有些相似,但是唯一不同的是此方式的并行发生在同一个设备上,而不是在不同的设备之间。并且在计算一批简单的样例时,允许进行“填充间隙”,这可以充分利用空闲的设备资源。

关于Tensorflow分布式并行策略

来源:https://blog.csdn.net/u010510350/article/details/53412659

标签:Tensorflow,分布式,并行
0
投稿

猜你喜欢

  • 解决MybatisPlus SqlServer OFFSET 分页问题

    2024-01-12 16:26:24
  • 使用Python解析Chrome浏览器书签的示例

    2023-03-12 01:14:34
  • Python持续监听文件变化代码实例

    2021-10-20 06:19:19
  • MySQL数据库的索引原理与慢SQL优化的5大原则

    2024-01-18 14:46:01
  • asp.net DropDownList实现二级联动效果

    2023-07-23 07:48:41
  • 一个图片后加载的代码

    2008-09-28 13:03:00
  • Go routine调度详解

    2024-05-10 10:57:57
  • 一些相见恨晚的 JavaScript 技巧

    2024-04-18 10:10:53
  • 在JavaScript中,为什么要尽可能使用局部变量?

    2009-03-01 12:38:00
  • 如何防范利用刷新来“作弊”的计数器?

    2009-11-22 17:18:00
  • 如何解决tensorflow恢复模型的特定值时出错

    2023-12-22 14:59:36
  • Python selenium根据class定位页面元素的方法

    2021-04-03 08:17:39
  • pytorch 查看cuda 版本方式

    2022-07-16 23:43:05
  • 对Python进行数据分析_关于Package的安装问题

    2021-07-29 17:23:42
  • DHTML+XML+ASP+CSS的树形目录

    2008-10-24 14:29:00
  • 在 mysql 中 定义 数据库,表,列时,设定的各个的编码格式

    2010-11-11 11:58:00
  • python sys模块使用方法介绍

    2021-11-24 10:02:11
  • 不用script仅用css编写无限级弹出菜单

    2008-04-24 14:03:00
  • 浅谈Go语言中的次方用法

    2024-02-17 04:57:08
  • sql server 创建临时表的使用说明

    2024-01-16 18:30:23
  • asp之家 网络编程 m.aspxhome.com