【技术实现步骤摘要】
硬件参数值的确定方法、装置、电子设备及存储介质
[0001]本申请实施例涉及神经网络
,特别涉及一种硬件参数值的确定方法
、
装置
、
电子设备及存储介质
。
技术介绍
[0002]神经网络模型需要依附于神经网络芯片,而神经网络芯片一般包括多个硬件参数,而多个硬件参数的参数值会影响到神经网络模型的性能;因此,在训练神经网络模型之前,需要确定出神经网络芯片的多个硬件参数的参数值,多个硬件参数可以包括内存空间
、
支持的传输带宽等
。
[0003]相关技术中,基于神经网络芯片的多个硬件参数,确定多个参数值组合,参数值组合中包括多个硬件参数的参数值;基于多个参数值组合,通过架构仿真模型,确定神经网络模型的多个性能,架构仿真模型用于仿真神经网络模型;基于多个性能,确定神经网络芯片的目标参数值组合,目标参数值组合包括多个硬件参数的目标参数值,也即最终确定出的参数值
。
[0004]由于硬件参数的种类较多,且硬件参数的参数值较多,因此,穷举这些硬件参数的参数值,就会导致得到大量的参数值组合,从而基于大量的参数值组合确定神经网络模型的性能会花费较多时间,进而导致确定硬件参数值的效率较低
。
技术实现思路
[0005]本申请实施例提供了一种硬件参数值的确定方法
、
装置
、
电子设备及存储介质,能够提高确定硬件参数值的效率
。
所述技术方案如下:
[0006]一
【技术保护点】
【技术特征摘要】
1.
一种硬件参数值的确定方法,其特征在于,所述方法包括:确定神经网络芯片的多个目标硬件参数;确定所述多个目标硬件参数的参数值变化范围;基于所述多个目标硬件参数的参数值变化范围,确定多个参数优化函数,一个参数优化函数对应一个硬件参数的参数值变化范围,且所述参数优化函数用于限制所述目标硬件参数在所述参数值变化范围内变化;基于所述多个参数优化函数,限制架构仿真模型在所述多个目标硬件参数的参数值变化范围内迭代训练,确定所述多个目标硬件参数的目标参数值,所述架构仿真模型用于仿真所述神经网络芯片的架构,所述目标硬件参数的目标参数值在所述硬件参数的参数值变化范围内
。2.
根据权利要求1所述的方法,其特征在于,所述基于所述多个参数优化函数,限制架构仿真模型在所述多个目标硬件参数的参数值变化范围内迭代训练,确定所述多个目标硬件参数的目标参数值,包括:基于所述多个参数优化函数,在所述多个目标硬件参数的参数值变化范围内确定第一参数值组合,所述第一参数值组合包括所述多个目标硬件参数的第一参数值;基于所述架构仿真模型,对所述第一参数值组合进行迭代训练;在所述第一参数值组合满足迭代收敛条件的情况下,将所述第一参数值组合中的多个目标硬件参数的第一参数值作为所述多个目标硬件参数的目标参数值
。3.
根据权利要求2所述的方法,其特征在于,所述方法还包括:在所述第一参数值组合不满足所述迭代收敛条件的情况下,基于所述多个参数优化函数,在所述多个目标硬件参数的参数值变化范围内确定第二参数值组合,所述第二参数值组合包括所述多个目标硬件参数的第二参数值;基于所述架构仿真模型,对所述第二参数值组合进行迭代训练,直到获取到满足所述迭代收敛条件的多个目标硬件参数的目标参数值为止
。4.
根据权利要求2所述的方法,其特征在于,所述基于所述多个参数优化函数,在所述多个目标硬件参数的参数值变化范围内确定第一参数值组合,包括:确定所述多个目标硬件参数的性能影响参数,所述硬件参数的性能影响参数用于表示所述硬件参数的参数值变化对所述神经网络芯片性能的影响程度和趋势;基于所述多个目标硬件参数的性能影响参数和所述多个参数优化函数,在多个目标硬件参数的参数值变化范围内确定第一参数值组合,所述第一参数值组合包括的多个目标硬件参数的参数值使得所述神经网络芯片的性能满足需求,且所述多个目标硬件参数的参数值在所述多个参数优化函数限制的参数值变化范围内
。5.
根据权利要求1所述的方法,其特征在于,所述方法还包括:获取性能需求信息,所述性能需求信息用于表示所述神经网络芯片所需满足的性能需求;基于所述性能需求信息,确定所述神经网络芯片的需求优化函数;所述基于所述多个参数优化函数,限制架构仿真模型在所述多个目标硬件参数的参数值变化范围...
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。