模型代码转换方法技术

本公开提供了一种模型代码转换方法

【技术实现步骤摘要】
模型代码转换方法、装置、设备及存储介质


[0001]本公开涉及计算机
，尤其涉及深度学习
、
人工智能等
，具体涉及一种模型代码转换方法
、
装置
、
设备及存储介质
。

技术介绍

[0002]随深度学习领域快速发展，大量创新模型算法喷涌而出，其中很多模型算法的代码都会基于各种不同的深度学习平台进行开发
。
而不同的深度学习平台存在差异，所以同样的模型算法基于不同深度学习平台开发得到的代码就不同，如何将模型代码快速在不同的深度学习平台应用起来也成为研究的热点
。

技术实现思路

[0003]本公开提供了一种模型代码转换方法
、
装置
、
设备及存储介质
。
[0004]根据本公开的一方面，提供了一种模型代码转换方法，包括：
[0005]对模型源代码进行第一解析处理，得到第一抽象语法树，其中，模型源代码为基于第一深度学习平台得到的；
[0006]对第一抽象语法树进行转换处理，得到本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.
一种模型代码转换方法，包括：对模型源代码进行第一解析处理，得到第一抽象语法树，其中，所述模型源代码为基于第一深度学习平台得到的；对所述第一抽象语法树进行转换处理，得到与第二深度学习平台对应的第二抽象语法树；对所述第二抽象语法树进行第一逆解析处理，得到目标模型代码，所述目标模型代码用于所述第二深度学习平台
。2.
根据权利要求1所述的方法，其中，所述第一抽象语法树包括多个第一节点，对所述第一抽象语法树进行转换处理，得到与第二深度学习平台对应的第二抽象语法树，包括：针对所述多个第一节点中的第一节点，获取所述第一节点中的第一接口函数，其中，所述第一接口函数为所述第一深度学习平台的接口函数；基于预设转换规则，将所述第一接口函数转换为与所述第一接口函数对应的第二接口函数，得到所述第一节点对应的第二节点，其中，所述第二接口函数为所述第二深度平台的接口函数，且所述第二节点为所述第二抽象语法树中的节点；至少基于所述第二节点，确定所述第二抽象语法树
。3.
根据权利要求2所述的方法，其中，基于预设转换规则，将所述第一接口函数转换为与所述第一接口函数对应的第二接口函数，包括：对所述第一接口函数进行第二解析处理，得到所述第一接口函数的第一名称信息
、
所述第一接口函数的第一参数信息以及所述第一接口函数的第一参数用法信息；基于所述预设转换规则
、
所述第一名称信息
、
所述第一参数信息以及所述第一参数用法信息，确定所述第二接口函数的第二名称信息以及所述第二接口函数的第二参数信息；基于所述第二名称信息和所述第二参数信息，确定所述第二接口函数
。4.
根据权利要求1所述的方法，其中，对所述第一抽象语法树进行转换处理，得到与第二深度学习平台对应的第二抽象语法树，包括：利用生成式大模型对所述第一抽象语法树进行转换处理，得到与第二深度学习平台对应的所述第二抽象语法树
。5.
根据权利要求1‑4中任一项所述的方法，在对所述第二抽象语法树进行第一逆解析处理，得到目标模型代码之后，所述方法还包括：利用所述第二深度学习平台执行所述目标模型代码，得到与所述目标模型代码相对应的目标模型
。6.
根据权利要求1‑5中任一项所述的方法，在对所述第二抽象语法树进行第一逆解析处理，得到目标模型代码之后，所述方法还包括：将所述目标代码存储于目标地址，其中，所述第一深度学习平台
、
所述第二深度学习平台以及所述目标地址为基于使用对象的选取操作得到
。7.
一种模型代码转换装置，包括：第一解析单元，用于对模型源代码进行第一解析处理，得到第一抽象语法树，其中，所述模型源代码为基于第一深度学习平台得到的；转换单元，用于对所述第一抽象语法树进行转换处...

【专利技术属性】
技术研发人员：高翔，周威，胡晓光，
申请(专利权)人：北京百度网讯科技有限公司，
类型：发明
国别省市：