[发明专利]一种模型训练方法和相关装置在审
申请号: | 202210427939.0 | 申请日: | 2022-04-22 |
公开(公告)号: | CN115114927A | 公开(公告)日: | 2022-09-27 |
发明(设计)人: | 弓静 | 申请(专利权)人: | 腾讯科技(深圳)有限公司 |
主分类号: | G06F40/30 | 分类号: | G06F40/30;G06F40/289;G06N3/04;G06N3/08 |
代理公司: | 深圳市深佳知识产权代理事务所(普通合伙) 44285 | 代理人: | 林志鹏 |
地址: | 518057 广东省深圳*** | 国省代码: | 广东;44 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本申请实施例公开了一种模型训练方法和相关装置,至少涉及人工智能模型中的机器学习,确定待训练模型包括的m个张量与n个并行进程之间的对应关系,m个张量包括在n个张量集合中,每个张量集合包括m个张量中的部分张量,n个张量集合与n个并行进程的对应关系为一一对应关系,使得每个并行进程只维护部分张量。目标并行进程与目标张量具有对应关系,在进行迭代的过程中,目标并行进程仅基于目标张量更新待训练模型的参数,根据更新后的参数训练待训练模型。不仅降低了创建临时缓存的数量,还降低了临时缓存的频繁创建和释放产生的内存碎片。由此,通过每个并行进程至维护部分张量,降低了激活层内存、临时缓存等,进而降低了模型的显存占用。 | ||
搜索关键词: | 一种 模型 训练 方法 相关 装置 | ||
【主权项】:
暂无信息
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于腾讯科技(深圳)有限公司,未经腾讯科技(深圳)有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/202210427939.0/,转载请声明来源钻瓜专利网。
- 上一篇:一种电动汽车双主体协同优化方法及装置
- 下一篇:一种药材粉碎研磨一体装置