[发明专利]显存管理、模型训练方法、装置、电子设备及存储介质在审
| 申请号： | 202110150321.X | 申请日： | 2021-02-03 |
| 公开（公告）号： | CN112882830A | 公开（公告）日： | 2021-06-01 |
| 发明（设计）人： | 邓哲也;章玄润;高华佐 | 申请（专利权）人： | 北京迈格威科技有限公司 |
| 主分类号： | G06F9/50 | 分类号： | G06F9/50;G06N3/08;G06T1/20;G06T1/60 |
| 代理公司： | 北京润泽恒知识产权代理有限公司 11319 | 代理人： | 王婷婷 |
| 地址： | 100086 北京市海淀区科*** | 国省代码： | 北京;11 |
| 权利要求书： | 查看更多 | 说明书： | 查看更多 |
| 摘要： | 本发明公开一种显存管理、模型训练方法、装置、电子设备及存储介质，该显存管理方法包括：获取模型的当前轮训练对应的显存阈值；在所述电子设备的显存占用值大于所述显存阈值的情况下，确定满足张量选择规则的目标张量；释放所述目标张量占用的显存。可见，实施上述方法，可以通过释放显存中不影响模型正常训练的张量，来减少GPU显存占用，而不需要提前获得整张计算图的全局信息，做到了完全动态，从而实现动态图机制的深度学习框架下的显存管理。 | ||
| 搜索关键词： | 显存 管理 模型 训练 方法 装置 电子设备 存储 介质 | ||
【主权项】：
暂无信息
下载完整专利技术内容需要扣除积分，VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息，商用须获得专利权人授权。该专利全部权利属于北京迈格威科技有限公司，未经北京迈格威科技有限公司许可，擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作，请联系【客服】
本文链接：http://www.vipzhuanli.com/patent/202110150321.X/，转载请声明来源钻瓜专利网。
- 上一篇：一种太阳能发电站运动控制装置
- 下一篇：一种地图的显示方法和装置
