[发明专利]基于GPU的卷积神经网络批归一化处理方法在审
申请号: | 202110294597.5 | 申请日: | 2021-03-19 |
公开(公告)号: | CN113011563A | 公开(公告)日: | 2021-06-22 |
发明(设计)人: | 孙圣力;马建伟 | 申请(专利权)人: | 北京大学 |
主分类号: | G06N3/04 | 分类号: | G06N3/04;G06N3/08;G06T1/20 |
代理公司: | 无锡永乐唯勤专利代理事务所(普通合伙) 32369 | 代理人: | 孙际德 |
地址: | 100000*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本发明提供了一种基于GPU的卷积神经网络批归一化处理方法,包括:获取训练数据集,训练数据集包括多个与多个训练通道一一对应的训练数据子集;判断训练数据子集的数据规模是否超过预定阈值。在训练数据子集的数据规模未超过预定阈值时,训练数据子集被分配并缓存至GPU的一个计算单元上,计算单元对分配到的训练数据子集进行批归一化;在训练数据子集的数据规模超过预定阈值时,训练数据子集被分配并缓存至所述GPU的至少两个计算单元上,至少两个计算单元对分配到的训练数据子集进行批归一化。本发明中,训练数据被缓存在GPU的计算单元的片上高速存储中,因此BN处理过程中,GPU无需再执行重复的数据读取过程,从而提升了BN的处理效率。 | ||
搜索关键词: | 基于 gpu 卷积 神经网络 归一化 处理 方法 | ||
【主权项】:
暂无信息
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京大学,未经北京大学许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/202110294597.5/,转载请声明来源钻瓜专利网。
- 上一篇:一种囊膜式分子弹簧隔振缓冲器
- 下一篇:一种电网电压暂降仿真方法