首页 专利交易 科技果 科技人才 科技服务 商标交易 会员权益 IP管家助手 需求市场 关于龙图腾
 /  免费注册
到顶部 到底部
清空 搜索

自动显存调优的模型并行训练方法、装置及设备 

申请/专利权人:清华大学

申请日:2024-03-28

公开(公告)日:2024-07-05

公开(公告)号:CN118297113A

主分类号:G06N3/0455

分类号:G06N3/0455;G06N3/084;G06F9/50

优先权:

专利状态码:在审-实质审查的生效

法律状态:2024.07.23#实质审查的生效;2024.07.05#公开

摘要:本发明涉及深度学习领域,公开了一种自动显存调优的模型并行训练方法、装置及设备,该方法包括:将显存调优方案与模型并行训练框架结合;将显存调优方案的通信时间与模型的计算时间进行重叠处理;在重叠处理后,确定模型的开关选择方案,基于不同开关选择方案对模型进行性能预测;基于动态规划算法,根据性能预测结果和机器显存容量调优,确定最优开关选择方案。本发明解决了如何根据硬件资源条件,动态调优选择最优的训练策略,同时仍保持较高速的训练速度问题。

主权项:1.一种自动显存调优的模型并行训练方法,其特征在于,包括:将显存调优方案与模型并行训练框架结合;将所述显存调优方案的通信时间与模型的计算时间进行重叠处理;在重叠处理后,确定模型的开关选择方案,基于不同开关选择方案对模型进行性能预测;基于动态规划算法,根据性能预测结果和机器显存容量调优,确定最优开关选择方案。

全文数据:

权利要求:

百度查询: 清华大学 自动显存调优的模型并行训练方法、装置及设备

免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。