买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
申请/专利权人:亿铸科技(上海)有限公司
摘要:本公开提供了一种学习模型的硬件部署方法、相关装置和介质。该硬件部署方法包括:获取用于目标学习模型的处理单元部署的初始流水并行数和初始张量并行数;在部署方案节点跳转关系图上,定位初始流水并行数和初始张量并行数所在的初始部署方案节点;将部署方案节点跳转关系图和初始部署方案节点输入强化学习模型,得到使目标学习模型的推理性能最优化的目标部署方案节点,目标部署方案节点包括目标流水并行数和目标张量并行数;基于目标流水并行数和目标张量并行数,为目标学习模型部署处理单元。本公开实施例能扩大部署方法的适用范围,提高部署效率。
主权项:1.一种学习模型的硬件部署方法,其特征在于,包括:获取用于目标学习模型的处理单元部署的初始流水并行数和初始张量并行数;在部署方案节点跳转关系图上,定位所述初始流水并行数和所述初始张量并行数所在的初始部署方案节点,其中,所述部署方案节点跳转关系图的每个部署方案节点包括流水并行数和张量并行数;将所述部署方案节点跳转关系图和所述初始部署方案节点输入强化学习模型,得到使所述目标学习模型的推理性能最优化的目标部署方案节点,所述目标部署方案节点包括目标流水并行数和目标张量并行数,其中,所述强化学习模型通过为多个样本方案节点跳转关系图中的每个样本方案节点跳转关系图,指定推理性能最优化的标签部署方案节点,并从所述样本方案节点跳转关系图的样本初始部署方案节点开始,按照奖励值最大化的原则跳转,直至跳转到所述标签部署方案节点的方式训练;基于所述目标流水并行数和所述目标张量并行数,为所述目标学习模型部署处理单元。
全文数据:
权利要求:
百度查询: 亿铸科技(上海)有限公司 学习模型的硬件部署方法、相关装置和介质
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。