一种模型调试方法、装置、电子设备及存储介质与流程

文档序号:33507526发布日期:2023-03-18 08:58阅读:来源:国知局

技术特征:
1.一种模型调试方法,包括:提取网络模型中的至少一个组件;以动态图模式加载调试工具;将所述至少一个组件对应的算子构成信息与所述调试工具中的算子融合策略进行匹配处理,得到匹配结果;以及,根据所述匹配结果,得到用于模型调试的调试信息。2.根据权利要求1所述的方法,还包括:构建所述网络模型的融合算子映射表。3.根据权利要求2所述的方法,其中,所述融合算子映射表包括:所述至少一个组件对应的算子构成信息。4.根据权利要求3所述的方法,其中,所述将所述至少一个组件对应的算子构成信息与所述调试工具中的算子融合策略进行匹配处理,得到匹配结果,包括:从第一计算图中得到第一子图,其中,所述第一计算图用于表征与所述融合算子映射表对应的全量计算图,所述第一子图用于表征与所述算子构成信息对应的局部计算图;获取第二子图,所述第二子图为所述调试工具中用于表征局部融合算子的局部计算图;将所述第一子图与所述第二子图进行所述匹配处理,所述匹配结果为所述第一子图与所述第二子图为同构图的情况下,匹配成功。5.根据权利要求1-4中任一项所述的方法,其中,所述将所述至少一个组件对应的算子构成信息与所述调试工具中的算子融合策略进行匹配处理,得到匹配结果,包括:在所述网络模型中存在同类型的n个组件的情况下,将一个组件对应的算子构成信息与所述调试工具中的算子融合策略进行匹配处理,得到所述匹配结果;其中,所述n为大于1的整数;针对同类型的n-1个组件,不执行所述匹配处理。6.根据权利要求1-4中任一项所述的方法,还包括:采用统一的应用程序接口api接口,反馈所述调试信息;根据所述调试信息对所述网络模型进行模型优化,得到目标网络模型。7.根据权利要求6所述的方法,其中,所述调试信息包括:与融合算子相关的算子信息、与所述算子信息对应的组网代码范围。8.一种模型调试装置,包括:提取模块,用于提取网络模型中的至少一个组件;加载模块,用于以动态图模式加载调试工具;匹配模块,用于将所述至少一个组件对应的算子构成信息与所述调试工具中的算子融合策略进行匹配处理,得到匹配结果;调试模块,用于根据所述匹配结果,得到用于模型调试的调试信息。9.根据权利要求8所述的装置,还包括构建模块,用于:构建所述网络模型的融合算子映射表。10.根据权利要求9所述的装置,其中,所述融合算子映射表包括:所述至少一个组件对应的算子构成信息。
11.根据权利要求10所述的装置,其中,所述匹配模块,用于:从第一计算图中得到第一子图,其中,所述第一计算图用于表征与所述融合算子映射表对应的全量计算图,所述第一子图用于表征与所述算子构成信息对应的局部计算图;获取第二子图,所述第二子图为所述调试工具中用于表征局部融合算子的局部计算图;将所述第一子图与所述第二子图进行所述匹配处理,所述匹配结果为所述第一子图与所述第二子图为同构图的情况下,匹配成功。12.根据权利要求8-11中任一项所述的装置,其中,所述匹配模块,用于:在所述网络模型中存在同类型的n个组件的情况下,将一个组件对应的算子构成信息与所述调试工具中的算子融合策略进行匹配处理,得到所述匹配结果;其中,所述n为大于1的整数;针对同类型的n-1个组件,不执行所述匹配处理。13.根据权利要求8-11中任一项所述的装置,还包括接口调用模块,用于:采用统一的应用程序接口api接口,反馈所述调试信息;根据所述调试信息对所述网络模型进行模型优化,得到目标网络模型。14.根据权利要求13所述的装置,其中,所述调试信息包括:与融合算子相关的算子信息、与所述算子信息对应的组网代码范围。15.一种电子设备,包括:至少一个处理器;以及与所述至少一个处理器通信连接的存储器;其中,所述存储器存储有可被所述至少一个处理器执行的指令,所述指令被所述至少一个处理器执行,以使所述至少一个处理器能够执行权利要求1-7中任一项所述的方法。16.一种存储有计算机指令的非瞬时计算机可读存储介质,其中,所述计算机指令用于使所述计算机执行根据权利要求1-7中任一项所述的方法。17.一种计算机程序产品,包括计算机程序,所述计算机程序在被处理器执行时实现根据权利要求1-7中任一项所述的方法。

技术总结
本公开提供了一种模型调试方法、装置、电子设备及存储介质,涉及人工智能技术领域,尤其涉及深度学习、用户编码/编译等领域。具体实现方案为:提取网络模型中的至少一个组件,以动态图模式加载调试工具,将至少一个组件对应的算子构成信息与调试工具中的算子融合策略进行匹配处理,得到匹配结果,以及,根据匹配结果得到用于模型调试的调试信息。采用本公开,可以实现模型的调试,以优化模型的性能。以优化模型的性能。以优化模型的性能。


技术研发人员:郝宏翔 巩伟宝 沈亮 吴志华 于佃海
受保护的技术使用者:北京百度网讯科技有限公司
技术研发日:2022.08.24
技术公布日:2023/3/17
当前第2页1 2 
网友询问留言 已有6条留言
  • 131553... 来自[中国] 2023年03月20日 16:24
    做的很详细
    0
  • 191576... 来自[中国] 2023年03月20日 12:58
    6的
    1
  • 198525... 来自[中国] 2023年03月19日 16:52
    0
  • 177347... 来自[中国] 2023年03月19日 14:56
    做的真详细!!做的真详细!!做的真详细!!做的真详细!!
    0
  • 182022... 来自[中国] 2023年03月18日 17:09
    一种模型调试方法、装置、电子设备及存储介质与流程 
    0
  • 184381... 来自[中国] 2023年03月18日 10:17
    做的真详细!!
    1
1