摘要
本发明提供了一种模型转换方法、装置、电子设备及存储介质,该方法包括:获取预先转换的中间模型的模型信息,根据头信息生成模型结构描述文件,采用模型结构描述文件构建模型转换框架,将常量信息输入至模型转换框架,确定计算图信息中的算子结构以及算子类型,根据预设算子映射表和算子类型将算子结构映射为目标算子结构,将目标算子结构填充至模型转换框架,得到目标模型。本发明实施例利用算子映射关系,实现了Onnx模型到PyTorch模型的准确转换,保证转换后模型的可读性,满足不同类型前端框架模型在后端设备上的部署需求。
技术关键词
模型转换方法
框架
模型转换装置
上存储计算机程序
电子设备
可读存储介质
处理器
模块
指令
存储器
样本
关系
数据