[发明专利]一种分布式并行训练的方法、设备及可读介质在审
申请号: | 202010154752.9 | 申请日: | 2020-03-08 |
公开(公告)号: | CN111381966A | 公开(公告)日: | 2020-07-07 |
发明(设计)人: | 孙红岩 | 申请(专利权)人: | 苏州浪潮智能科技有限公司 |
主分类号: | G06F9/50 | 分类号: | G06F9/50;G06N20/00 |
代理公司: | 北京连和连知识产权代理有限公司 11278 | 代理人: | 刘小峰 |
地址: | 215100 江苏省苏州市吴*** | 国省代码: | 江苏;32 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本发明公开了一种分布式并行训练的方法,包括以下步骤:为多个进程分配GPU;将多个进程下的多个GPU设置为采用相同的优化器和网络状态参数;在多个进程下,将训练模型的不同网络层分配到不同的GPU上;以及将训练集分配到多个进程,并通过多个进程下的多个GPU并行进行训练。本发明还公开了一种计算机设备和可读存储介质。本发明通过将训练模型的不同训练层分配到不同的GPU上实现模型并行训练,并将数据分配给不同线程上实现数据并行训练,不同线程分配多个GPU进行训练,实现了模型并行和数据并行混合训练,提高数据处理效率。 | ||
搜索关键词: | 一种 分布式 并行 训练 方法 设备 可读 介质 | ||
【主权项】:
暂无信息
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于苏州浪潮智能科技有限公司,未经苏州浪潮智能科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/202010154752.9/,转载请声明来源钻瓜专利网。