[发明专利]一种基于深度强化学习的流水线并行训练任务分配方法在审
| 申请号： | 202110563603.2 | 申请日： | 2021-05-24 |
| 公开（公告）号： | CN113312178A | 公开（公告）日： | 2021-08-27 |
| 发明（设计）人： | 毛莺池;屠子健;吴俊;平萍;郭宏乐;徐淑芳 | 申请（专利权）人： | 河海大学 |
| 主分类号： | G06F9/50 | 分类号： | G06F9/50;G06N3/04;G06N3/08 |
| 代理公司： | 南京乐羽知行专利代理事务所(普通合伙) 32326 | 代理人： | 李玉平 |
| 地址： | 211100 江苏*** | 国省代码： | 江苏;32 |
| 权利要求书： | 查看更多 | 说明书： | 查看更多 |
| 摘要： | 本发明公开一种基于深度强化学习的流水线并行训练任务分配方法，步骤为：初始化深度学习模型，并把该模型载入；建立对目标深度学习模型运行时分析方案，构建模型参数文本库；根据获取的模型参数文本信息，构建任务分配预测网络；使用策略梯度训练预测网络，直至生成最优流水线并行任务分配方案；按照生成的最优任务分配方案，在异构计算节点中部署模型，完成训练任务。本发明有效避免节点计算能力和网络带宽差差异带来的负载不均衡问题，提高模型训练速度，并且模型参数越大，提升效果越明显。 | ||
| 搜索关键词： | 一种 基于 深度 强化 学习 流水线 并行 训练 任务 分配 方法 | ||
【主权项】：
暂无信息
下载完整专利技术内容需要扣除积分，VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息，商用须获得专利权人授权。该专利全部权利属于河海大学，未经河海大学许可，擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作，请联系【客服】
本文链接：http://www.vipzhuanli.com/patent/202110563603.2/，转载请声明来源钻瓜专利网。
