[发明专利]处理内存访问开销的方法、装置及介质有效
申请号: | 201910989917.1 | 申请日: | 2019-10-17 |
公开(公告)号: | CN110704197B | 公开(公告)日: | 2022-12-09 |
发明(设计)人: | 李寅;何亮亮 | 申请(专利权)人: | 北京小米移动软件有限公司 |
主分类号: | G06F9/50 | 分类号: | G06F9/50;G06N3/04;G06N3/08 |
代理公司: | 北京名华博信知识产权代理有限公司 11453 | 代理人: | 白莹 |
地址: | 100085 北京市海淀区清河*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 处理 内存 访问 开销 方法 装置 介质 | ||
1.处理内存访问开销的方法,其特征在于,应用于深度学习过程,所述方法包括:
申请用于针对深度学习模型中至少一卷积操作进行数值填补操作的内存;
确定所述深度学习模型的输入数据;
使用所述深度学习模型对所述输入数据进行深度学习处理;
释放所述内存;
所述申请用于针对深度学习模型中至少一卷积操作进行数值填补操作的内存,包括:
确定N个卷积操作的填补后数据的尺寸中的最大尺寸,N为大于1的整数,为每层卷积操作均设置一尺寸与此最大尺寸相同的内存;或者,
使多个满足尺寸不增大规律的卷积操作共享一个内存。
2.如权利要求1所述的方法,其特征在于,
所述深度学习模型包括按顺序设置的N个卷积操作,N为大于1的整数;
所述申请用于针对深度学习模型中至少一卷积操作进行数值填补操作的内存,包括:
确定所述深度学习模型中第一个卷积操作的输入数据的尺寸;
根据所述第一个卷积操作的输入数据的尺寸确定所述N个卷积操作的输出数据的尺寸和所述N个卷积操作的输入数据执行数值填补操作后的数据尺寸;
确定所述N个卷积操作的输入数据执行数值填补操作后的数据尺寸中的最大尺寸;
申请N个内存,所述N个内存与所述N个卷积操作一一对应,所述N个内存的容量相同并且均对应于所述最大尺寸;
所述使用所述深度学习模型对所述输入数据进行深度学习处理包括:
执行每次卷积操作前,使用所述卷积操作对应的内存对所述卷积操作的输入数据进行数值填补。
3.如权利要求1所述的方法,其特征在于,
所述深度学习模型包括按顺序设置的N个卷积操作,N为大于1的整数;
所述申请用于针对深度学习模型中至少一卷积操作进行数值填补操作的内存,包括:
确定所述深度学习模型中第一个卷积操作的输入数据的尺寸;
根据所述第一个卷积操作的输入数据的尺寸确定所述N个卷积操作的输出数据的尺寸和所述N个卷积操作的输入数据执行数值填补操作后的数据尺寸;
根据所述N个卷积操作的输入数据执行数值填补操作后的数据尺寸确定K个卷积操作集合,所述N个卷积操作由所述K个卷积操作集合组成,每个卷积操作集合包含的连续的卷积操作的输出数据的尺寸满足不增大规律;
申请K个内存,所述K个内存与所述K个卷积操作集合一一对应,所述K个内存的容量分别对应于所述K个卷积操作集合中第一个卷积操作的输入数据执行数值填补操作后的数据尺寸,所述K是小于N的整数;
所述使用所述深度学习模型对所述输入数据进行深度学习处理,包括:
在执行每次卷积操作前,使用所述卷积操作所属的卷积操作集合所对应的内存对所述卷积操作的输入数据进行数值填补。
4.如权利要求1所述的方法,其特征在于,
所述深度学习模型包括按顺序设置的N个卷积操作,N为大于1的整数;
所述申请用于针对深度学习模型中至少一卷积操作进行数值填补操作的内存,包括:
确定所述深度学习模型中第一个卷积操作的输入数据的尺寸;
根据所述第一个卷积操作的输入数据的尺寸确定所述N个卷积操作的输出数据的尺寸和所述N个卷积操作的输入数据执行数值填补操作后的数据尺寸;
根据K个卷积操作的输入数据执行数值填补操作后的数据尺寸确定K个卷积操作集合,所述N个卷积操作由所述K个卷积操作集合组成,每个卷积操作集合中包含的卷积操作的输入数据执行数值填补操作后的数据尺寸相同;
申请K个内存,所述K个内存与所述K个卷积操作集合一一对应,所述K个内存的容量分别对应于K个卷积操作集合中任一卷积操作的输入数据执行数值填补操作后的数据尺寸,K是小于或等于N的整数;
所述使用所述深度学习模型对所述输入数据进行深度学习处理,包括:
在执行每次卷积操作前,使用所述卷积操作所属的卷积操作集合所对应的内存对所述卷积操作的输入数据进行数值填补。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京小米移动软件有限公司,未经北京小米移动软件有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/201910989917.1/1.html,转载请声明来源钻瓜专利网。