神经网络搜索方法、装置和计算机设备
摘要:
本申请涉及一种神经网络搜索方法、装置和计算机设备。所述方法包括:确定目标网络层数;根据目标网络层数确定多个初始化神经网络模型,对各初始化神经网络模型进行表达能力判断,得到对应的判断结果;针对任一初始化神经网络模型,在判断结果表征初始化神经网络模型的表达能力满足预设条件的情况下,确定初始化神经网络模型的片内存储信息和片外存储信息,在初始化神经网络模型的片内存储信息和片外存储信息满足存储条件的情况下,根据目标带宽利用率确定初始化神经网络模型的性能数据;根据各初始化神经网络模型的性能数据,确定目标神经网络模型。采用本方法能够快速的获取到在部署芯片性能较好、精度较高的目标神经网络模型。
0/0