一种基于布局传播的机器学习模型转化方法、装置以及设备制造方法及图纸

技术编号:41308946 阅读:24 留言:0更新日期:2024-05-13 14:53
本说明书实施例公开了一种基于布局传播的机器学习模型转化方法、装置以及设备,采用了布局传播法,区别于传统的算子插入法。方案包括:将原始机器学习模型转化为框架转移中间模型;对框架转移中间模型进行解析,得到包含多个算子的计算图;在多个算子中确定具有隐式布局的算子;判断确定的算子的隐式布局是否与其输入张量的布局变量一致,布局变量是根据目标机器学习模型的目标布局为输入张量显式地赋予的;若否,则对确定的算子的属性进行调整,以使其适应于输入张量;根据调整后的算子,生成目标机器学习模型。

【技术实现步骤摘要】

本说明书涉及机器学习领域,尤其涉及一种基于布局传播的机器学习模型转化方法、装置以及设备


技术介绍

1、随着计算机和互联网技术的发展,出现了多种类型的机器学习框架,而不同的环境对于各类型机器学习框架的支持能力也不同,这就导致一些已经训练完的机器学习模型,难以直接投入到新的环境中使用,或者在直接投入后容易出现异常问题。

2、为解决该问题,可以通过能够适应新环境的其他机器学习框架重新构造相同的机器学习模型,手动将原始机器学习模型中的参数层权重进行移植。或者,也可以先将原始机器学习模型转化为中间格式的模型,通过引入冗余算子将该中间格式的模型进行转化,得到目标机器学习模型。

3、基于此,需要更能够保证目标机器学习模型的生成效率和运行效率的方案。


技术实现思路

1、本说明书一个或多个实施例提供一种基于布局传播的机器学习模型转化方法、装置、设备以及存储介质,用以解决如下技术问题:需要更能够保证目标机器学习模型的生成效率和运行效率的方案。

2、为解决上述技术问题,本说明书一个或多个实施例本文档来自技高网...

【技术保护点】

1.一种基于布局传播的机器学习模型转化方法,包括:

2.如权利要求1所述的方法,所述根据所述调整后的算子,生成所述目标机器学习模型,具体包括:

3.如权利要求1所述的方法,所述根据所述调整后的算子,生成所述目标机器学习模型,具体包括:

4.如权利要求3所述的方法,所述生成所述目标机器学习模型之后,所述方法还包括:

5.如权利要求4所述的方法,所述第一激活算子为LeakyReLU算子,所述第二激活算子为PReLU算子,所述第二卷积算子为融合卷积算子。

6.如权利要求4所述的方法,所述确定的算子包括卷积算子,所述移动端模型用于辅助用...

【技术特征摘要】

1.一种基于布局传播的机器学习模型转化方法,包括:

2.如权利要求1所述的方法,所述根据所述调整后的算子,生成所述目标机器学习模型,具体包括:

3.如权利要求1所述的方法,所述根据所述调整后的算子,生成所述目标机器学习模型,具体包括:

4.如权利要求3所述的方法,所述生成所述目标机器学习模型之后,所述方法还包括:

5.如权利要求4所述的方法,所述第一激活算子为leakyrelu算子,所述第二激活算子为prelu算子,所述第二卷积算子为融合卷积算子。

6.如权利要求4所述的方法,所述确定的算子包括卷积算子,所述移动端模型用于辅助用户拍摄图像。

7.如权利要求1所述的方法,所述对所述确定的算子的属性进行调整,具体包括:

8.如权利要求7所述的方法,所述隐式布局为nchw或nhwc,所述输入张量的布局变量为nchw或nhwc,所述隐式布局与所述输入张量的布局变量不一致。

9.如权利要求1~8任一项所述的方法,所述原始机器学习模型为pytorch模型,所述框架转移中间模型为onnx模型,所述目标机器学习模型为tensorflow模型。

10.一种基于布局传播的机器学习模型转化装置,包括:

11.如权利要求10...

【专利技术属性】
技术研发人员:黄堃程远王萌
申请(专利权)人:支付宝杭州信息技术有限公司
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1