发明公开
- 专利标题: 神经网络搜索方法、装置和计算机设备
-
申请号: CN202211423554.3申请日: 2022-11-15
-
公开(公告)号: CN115759197A公开(公告)日: 2023-03-07
- 发明人: 尹首一 , 韩振华 , 韩慧明 , 魏少军
- 申请人: 清华大学
- 申请人地址: 北京市海淀区清华园
- 专利权人: 清华大学
- 当前专利权人: 清华大学
- 当前专利权人地址: 北京市海淀区清华园
- 代理机构: 北京华进京联知识产权代理有限公司
- 代理商 安孔川
- 主分类号: G06N3/0464
- IPC分类号: G06N3/0464 ; G06N3/048 ; G06N3/08 ; G06F18/24 ; G06V10/82 ; G06V10/764
摘要:
本申请涉及一种神经网络搜索方法、装置和计算机设备。所述方法包括:确定目标网络层数;根据目标网络层数确定多个初始化神经网络模型,对各初始化神经网络模型进行表达能力判断,得到对应的判断结果;针对任一初始化神经网络模型,在判断结果表征初始化神经网络模型的表达能力满足预设条件的情况下,确定初始化神经网络模型的片内存储信息和片外存储信息,在初始化神经网络模型的片内存储信息和片外存储信息满足存储条件的情况下,根据目标带宽利用率确定初始化神经网络模型的性能数据;根据各初始化神经网络模型的性能数据,确定目标神经网络模型。采用本方法能够快速的获取到在部署芯片性能较好、精度较高的目标神经网络模型。