专利检索 ap:("上海燧原科技有限公司") AND inv:"王正俊" 第 1 页

1.

发明公开
一种训练图处理方法、装置、设备及存储介质审中-公开

公开(公告)号：CN116957108A

公开(公告)日：2023-10-27

申请号：CN202310923828.3

申请日：2023-07-26

申请人： 上海燧原科技有限公司

发明人： 万学磊 , 王正俊 , 田野

IPC分类号： G06N20/00

摘要： 本发明公开了一种训练图处理方法、装置、设备及存储介质。包括：获取训练图，根据预置的节点清单确定第一记录数据；根据训练图建立优化器的权重联动关系，根据权重联动关系确定第二记录数据；当检测到发送数据时，根据发送数据、第一记录数据和第二记录数据生成训练图处理结果。通过节点清单确定不需要插入转换算子的权重或权重梯度以生成第一记录数据，通过训练图建立优化器的权重联动关系，进而生成第二记录数据，当检测到发送数据时可以通过判断权重张量实现对第一记录数据和第二记录数据的修改，最终生成训练图处理结果，避免芯片在与第三方框架对接时因为格式不匹配而出现的格式转换算子数量过多，模型训练效率低的问题，提升了训练性能。