(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202210168582.9 (22)申请日 2022.02.23 (71)申请人 北京三快在线科技有限公司 地址 100080 北京市海淀区北四环西路9号 2106-030 (72)发明人 贺捷　吴望龙　 (74)专利代理 机构 北京曼威知识产权代理有限 公司 11709 专利代理师 邓超 (51)Int.Cl. G06F 16/178(2019.01) G06F 16/172(2019.01) G06F 9/445(2018.01) G06N 20/00(2019.01) (54)发明名称 一种模型参数转换的方法及装置 (57)摘要 本说明书公开了一种模型参数转换的方法 及装置， 可以获取待构建模型所需参数类型下的 模型参数， 并将该模型参数传入待构建模型中， 以通过该待构建模型中预先部署的参数转换关 系， 并基于该模型参数进行格式转换， 得到标准 格式下的第一参数， 进而， 将标准格式下的第一 参数， 通过待构建模型中预设的转换函数进行转 换， 得到自定义计算层所需参数类型下的第一参 数， 并传入到待构建模型的自定义计算层中， 以 构建得到包含有自定义计算层的待构建模型， 以 及通过构建出的待构建模型执行待构建模型对 应的目标业务， 其中， 自定义计算层是针对该目 标业务额外配置出的网络层， 从而提高了模型部 署的效率。 权利要求书2页 说明书11页 附图3页 CN 114546973 A 2022.05.27 CN 114546973 A 1.一种模型参数转换的方法， 其特 征在于， 包括： 获取待构建模型 所需参数类型 下的模型参数； 将所述模型参数传入所述待构建模型中， 以通过所述待构建模型中预先部署的参数转 换关系， 基于所述模型参数进行格式转换， 得到标准格式下的第一 参数； 将所述标准格式下的第一参数， 通过所述待构建模型中预设的转换函数进行转换， 得 到自定义计算层所需参数类型下 的第一参数， 并传入到所述待构建模型 的自定义计算层， 以构建得到包含有所述自定义计算层的待构建模型， 以及通过构建出的所述待构建模型执 行所述待构建模型对应的目标业务， 其中， 所述自定义计算层是针对所述 目标业务额外配 置出的网络层。 2.如权利要求1所述的方法， 其特征在于， 将所述模型参数传入所述待构建模型中， 以 通过所述待构建模型中预先部署的参数转换关系， 基于所述模型参数进行格式转换， 得到 标准格式下的第一 参数， 具体包括： 将所述模型参数传入所述待构建模型中， 以在所述待构建模型内部， 将所述模型参数 中包含的除第二参数外的第一参数进行格式转换， 得到所述标准格式下 的第一参数， 所述 第一参数包括所述自定义计算层内部进行模型计算所需的参数， 所述第二参数包括所述待 构建模型中除所述自定义计算层外的其 他网络结构所需的参数； 构建得到包 含有所述自定义计算层的待构建模型， 具体包括： 根据基于所述标准格式下的第 一参数， 转换得到的所述自定义计算层所需参数类型下 的第一参数， 以及所述第二 参数， 构建得到包 含有所述自定义计算层的待构建模型。 3.如权利要求1所述的方法， 其特 征在于， 所述方法还 包括： 在将所述待构建模型以文件格式进行存储时， 将所述标准格式下的第 一参数转换为字 节格式， 得到字节格式下 的第一参数， 以将所述字节格式下 的第一参数保存在所述待构建 模型所对应的模型文件中； 在通过所述模型文件加载所述待构建模型时， 将所述字节格式下的第 一参数还原为所 述标准格式下的第一参数， 以使 所述待构建模 型将所述标准格式下的第一参数通过所述转 换函数进行转换后， 传入到所述待构建模型的自定义计算层中。 4.如权利要求1～3任一项所述的方法， 其特征在于， 所述待构建模型至少包括第一模 型以及第二模型， 所述第一模型是在训练阶段以预设第一模型框架构建出的针对所述目标 业务的模型， 所述第二模型是通过训练后的第一模型在预设第二模型框架下构建出的用于 部署并实际执 行所述目标业 务的模型。 5.如权利要求4所述的方法， 其特征在于， 若所述待构建模型为第一模型， 获取待构建 模型所需的模型参数， 具体包括： 获取标准格式下的模型参数； 将所述标准格 式下的模型参数， 转换为所述第 一模型框架对应的第 一类型下的模型参 数， 并将所述第一类型 下的模型参数传入所述第一模型中； 若所述待构建模型为第二模型， 获取待构建模型 所需的模型参数， 具体包括： 将所述第一类型下的模型参数， 转换为所述第 二模型框架对应的第 二类型下的模型参 数， 并将所述第二类型 下的模型参数传入到所述第二模型中。 6.如权利要求5所述的方法， 其特 征在于， 获取 标准格式下的模型参数， 具体包括：权　利　要　求　书 1/2 页 2 CN 114546973 A 2获取预先配置的参数文件； 读取所述参数文件， 并将所述参数文件中存储的模型参数， 转换为所述标准格式下的 模型参数。 7.如权利 要求4所述的方法， 其特征在于， 所述第一模型框架为Torch框架， 所述第二模 型框架为TensorRT框架。 8.一种模型参数转换的装置， 其特 征在于， 包括： 获取模块， 用于获取待构建模型 所需参数类型 下的模型参数； 传入模块， 用于将所述模型参数传入所述待构建模型中， 以通过所述待构建模型中预 先部署的参数转换关系， 基于所述模型参数进行格式转换， 得到标准格式下的第一 参数； 转换模块， 用于将所述标准格式下的第一参数， 通过所述待构建模型中预设的转换函 数进行转换， 得到 自定义计算层所需参数类型下 的第一参数， 并传入到所述待构建模型 的 自定义计算层， 以构建得到包含有所述自定义计算层的待构建模型， 以及通过构建出 的所 述待构建模型执行所述待构建模型对应的目标业务， 其中， 所述自定义计算层是针对所述 目标业务额外配置出的网络层。 9.一种计算机可读存储介质， 其特征在于， 所述存储介质存储有计算机程序， 所述计算 机程序被处 理器执行时实现上述权利要求1～7任一项所述的方法。 10.一种电子设备， 包括存储器、 处理器及存储在存储器上并可在处理器上运行的计算 机程序， 其特征在于， 所述处理器执行所述程序时实现上述权利要求1～7任一项所述的方 法。权　利　要　求　书 2/2 页 3 CN 114546973 A 3