[发明专利]一种模型调试方法、装置、电子设备及存储介质

权利要求书

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个组件，不执行所述匹配处理。