发明 2020101167055 神经网络模型的结构搜索方法和装置 2024
已下证 人工智能 深度神经网络 数据处理 图像数据 视频处理 1人
免责声明:以上消息未经人工确认,本平台不担保其真实性和有效性,交易前请仔细核实。
本公开涉及人工智能领域。本公开的实施例公开了神经网络模型的结构搜索方法和装置。该方法包括:采用控制器从预设的网络结构搜索空间中搜索出候选网络结构;利用YUV图像训练数据训练候选网络结构;获取训练完成的候选网络结构的性能信息,根据训练完成的候选网络结构的性能信息生成反馈信息,并将反馈信息反馈至控制器,以使控制器基于反馈信息进行更新后从网络结构搜索空间中搜索出新的候选网络结构;响应于确定基于当前训练完成的候选网络结构的性能信息生成的反馈信息达到预设的收敛条件,基于当前训练完成的候选网络结构生成用于YUV图像数据的神经网络模型结构。该方法可以自动搜索出适合处理YUV图像数据的神经网络模型结构。