深度学习模型的模型结构保存方法、构建方法及相关装置

专利申请号：CN202211625085.3

公 开 号：CN116187412A

发 明 人：赖铖 董豪 韩佳容 

代 理 人：深圳市君胜知识产权代理事务所(普通合伙)王永文;陈俊贤

代理机构：深圳市君胜知识产权代理事务所(普通合伙)

专利类型：发明专利

申 请 日：20230530

公 开 日：20221216

专利主分类号：G06N3/08

关 键 词：网络层 学习 连接关系 模型结构 前向传播 存储 模型结构文件 网络层参数 网络层信息 标识信息 格式文件 相关装置 保存 构建 部署 统一 

摘      要：本发明公开了一种深度学习模型的模型结构保存方法、构建方法及相关装置，通过控制深度学习模型进行前向传播，并获取深度学习模型在前向传播过程中每一层网络层的网络层信息；根据每一层网络层的网络层标识信息，确定深度学习模型的网络层连接关系；将每一层网络层的网络层参数以及网络层连接关系存储为HDF5格式文件，得到深度学习模型的模型结构文件并进行存储。通过上述方案，可以实现对不同深度学习框架下的深度学习模型的模型结构进行统一保存，以便后续可直接部署在不同深度学习框架中。