[发明专利]一种基于ONNX面向Caffe2训练的深度学习模型自动转换方法在审
申请号: | 202110452694.2 | 申请日: | 2021-04-26 |
公开(公告)号: | CN114219083A | 公开(公告)日: | 2022-03-22 |
发明(设计)人: | 彭超;刘鑫;刘沙;陈德训;黄则强;高捷;王宜鹏 | 申请(专利权)人: | 无锡江南计算技术研究所 |
主分类号: | G06N3/08 | 分类号: | G06N3/08;G06N3/04 |
代理公司: | 苏州创元专利商标事务所有限公司 32103 | 代理人: | 王健 |
地址: | 214038 江苏*** | 国省代码: | 江苏;32 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本发明公开一种基于ONNX面向Caffe2训练的深度学习模型自动转换方法,包括以下步骤:S1、ONNX文件向Caffe2文件的映射,得到Caffe2格式的前向传播网络,S2、根据S1中获得的前向传播网络的信息,自动生成求梯度算子,得到反向传播网络,S3、根据满足生成完整反向传播网络的需要,设计用户配置信息格式,将配置信息融入上步骤的模型定义文件中。本发明可以自动产生包括正向传播神经网络、反向传播神经网络、运行控制块和辅助算子的模型文件,从而进行训练任务。 | ||
搜索关键词: | 一种 基于 onnx 面向 caffe2 训练 深度 学习 模型 自动 转换 方法 | ||
【主权项】:
暂无信息
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于无锡江南计算技术研究所,未经无锡江南计算技术研究所许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/202110452694.2/,转载请声明来源钻瓜专利网。