【技术实现步骤摘要】
用于确定深度神经网络的耗时的方法和装置
本申请实施例涉及计算机
,具体涉及计算机视觉
,尤其涉及用于确定深度神经网络的耗时的方法和装置。
技术介绍
随着深度学习技术的发展,该技术逐渐应用于越来越多的场景中,并且起到了举足轻重的作用。例如行人检测、无人驾驶、智慧零售等场景,都依赖于深度学习来完成特定任务。目前产业结合深度学习已是大势所趋,一方面能够提升效果,另一方面能够降低人工成本。每种进行深度学习的神经网络都有相应的耗时,为了能够在应用中对深度神经网络进行更好的控制,可以对深度神经网络的耗时进行测算。
技术实现思路
提供了一种用于确定深度神经网络的耗时的方法、装置、电子设备以及存储介质。根据第一方面,提供了一种用于确定深度神经网络的耗时的方法,包括:确定待处理深度神经网络包括的处理层,以及获取所确定的处理层的参数的值;在预设的运算耗时数据集中,查找所确定的处理层的参数的值所对应的运算耗时,作为所确定的处理层的运算耗时,其中,运算耗时数据集中包括对应存储的值与参数为该值的处理层的运算耗时;基于各个所确定的处理层的运算耗时,确定待处理深度神经网络的运算耗时。根据第二方面,提供了一种用于确定深度神经网络的耗时的装置,包括:获取单元,被配置成确定待处理深度神经网络包括的处理层,以及获取所确定的处理层的参数的值;查找单元,被配置成在预设的运算耗时数据集中,查找所确定的处理层的参数的值所对应的运算耗时,作为所确定的处理层的运算耗时,其中,运算耗时数据集中包括对应存储的值与参 ...
【技术保护点】
1.一种用于确定深度神经网络的耗时的方法,所述方法包括:/n确定待处理深度神经网络包括的处理层,以及获取所确定的处理层的参数的值;/n在预设的运算耗时数据集中,查找所确定的处理层的参数的值所对应的运算耗时,作为所确定的处理层的运算耗时,其中,所述运算耗时数据集中包括对应存储的值与参数为该值的处理层的运算耗时;/n基于各个所确定的处理层的运算耗时,确定所述待处理深度神经网络的运算耗时。/n
【技术特征摘要】
1.一种用于确定深度神经网络的耗时的方法,所述方法包括:
确定待处理深度神经网络包括的处理层,以及获取所确定的处理层的参数的值;
在预设的运算耗时数据集中,查找所确定的处理层的参数的值所对应的运算耗时,作为所确定的处理层的运算耗时,其中,所述运算耗时数据集中包括对应存储的值与参数为该值的处理层的运算耗时;
基于各个所确定的处理层的运算耗时,确定所述待处理深度神经网络的运算耗时。
2.根据权利要求1所述的方法,其中,所述运算耗时数据集通过以下步骤得到:
根据深度神经网络的处理层的待赋值运算公式,生成与该处理层的参数的多个值分别对应的、该处理层的多个运算公式;
获取执行所述多个运算公式分别对应的运算的运算耗时,其中,所确定的每个运算耗时为利用相同运算公式进行至少两次运算的运算耗时的平均值;
对于所述多个运算公式中的运算公式,将该运算公式中的参数的值和该运算公式对应的运算耗时,对应存储于运算耗时数据集。
3.根据权利要求2所述的方法,其中,深度神经网络包括不同种的处理层;所述根据深度神经网络的处理层的待赋值运算公式,生成与该处理层的参数的多个值分别对应的、该处理层的多个运算公式,包括:
根据每种处理层的待赋值运算公式,生成与每种处理层的参数的多个值分别对应的多个运算公式,其中,所述不同种的处理层包括:卷积层、池化层和全连接层。
4.根据权利要求3所述的方法,其中,所述运算耗时数据集包括与所述不同种的处理层分别对应的不同的运算耗时子集;
所述对于所述多个运算公式中的运算公式,将该运算公式中的参数的值和该运算公式对应的运算耗时,对应存储于运算耗时数据集,包括:
对于每种处理层的参数的多个值分别对应的多个运算公式中的运算公式,将该运算公式中的参数的值和该运算公式对应的运算耗时,对应存储于该种处理层对应的运算耗时子集,其中,所述不同的运算耗时子集包括卷积运算耗时子集、池化运算耗时子集和全连接运算耗时子集。
5.根据权利要求4所述的方法,其中,所述在预设的运算耗时数据集中,查找所确定的处理层的参数的值所对应的运算耗时,作为所确定的处理层的运算耗时,包括:
对于每个所确定的处理层,基于该处理层的种类,将所述运算耗时数据集的运算耗时子集中,该处理层对应的运算耗时子集作为与该处理层对应的目标运算耗时子集;
在所述目标运算耗时子集中,查找该处理层的参数的值所对应的运算耗时。
6.根据权利要求2所述的方法,其中,所述运算耗时数据集与指定电子设备相关联;
所述获取执行所述多个运算公式分别对应的运算的运算耗时,包括:
获取所述指定电子设备执行所述多个运算公式分别对应的运算的运算耗时;以及
所述基于各个所确定的处理层的运算耗时,确定所述待处理深度神经网络的运算耗时,包括:
基于各个所确定的处理层的运算耗时,确定所述指定电子设备执行所述待处理深度神经网络中的运算的运算耗时。
7.一种用于确定深度神经网络的耗时的装置,所述装置包括:
获取单元,被配置成确定待处理深度神经网络包括的处理层,以及获取所确定的处理层的参数的值;
查找单元,被配置成在预设的运算耗时数据集中,查找所确定的处理层的参数的值所对应的运算...
【专利技术属性】
技术研发人员:苑鹏程,韩树民,冯原,辛颖,
申请(专利权)人:北京百度网讯科技有限公司,
类型:发明
国别省市:北京;11
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。