摘要
本申请提出一种神经网络模型迁移精度对齐方法、装置、计算机设备,该方法包括:在将神经网络模型从第一框架迁移至第二框架时,获取第一框架中神经网络模型的第一计算图,获取第二框架中神经网络模型的第二计算图;针对任一计算图,根据计算图中多个节点之间的依赖关系从多个节点中筛选出待检查的目标节点;将第一计算图中第一目标节点的第一输入数据和第一输出数据以及第二计算图中第二目标节点的第二输入数据和第二输出数据进行对比分析,得到对比分析结果;根据对比分析结果从第二计算图的多个节点中筛选出异常节点。本申请能够显著提高跨框架迁移过程中的精度比对效率和精确度,自动化地检测和定位精度差异,为框架迁移提供强有力的支持。
技术关键词
神经网络模型
节点
数据
对齐方法
框架
监测点
计算机设备
关系
指标
对齐装置
精度
注意力
存储器
处理器
序列
模块
线性
指令