- 专利标题: 一种基于LSTM网络的片上微调方法及装置
-
申请号: CN202310147794.3申请日: 2023-02-22
-
公开(公告)号: CN116029332B公开(公告)日: 2023-08-22
- 发明人: 林军 , 周李鑫 , 崔峻伟 , 王中风
- 申请人: 南京大学
- 申请人地址: 江苏省南京市栖霞区仙林大道163号
- 专利权人: 南京大学
- 当前专利权人: 南京大学
- 当前专利权人地址: 江苏省南京市栖霞区仙林大道163号
- 代理机构: 北京弘权知识产权代理有限公司
- 代理商 逯长明; 占园
- 主分类号: G06N3/0442
- IPC分类号: G06N3/0442 ; G06N3/048 ; G06N3/084 ; G06N3/063
摘要:
本申请提供一种基于LSTM网络的片上微调方法及装置,方法包括:根据LSTM网络模型的参数,生成计算参数;根据计算参数对输入值和权重值进行细粒度切分,以及生成指令集合;在神经网络处理器NPU读取指令集合;根据指令集合执行NPU片上计算,以生成中间值和LSTM网络输出值;将LSTM网络输出值传输至误差函数,并通过误差函数得到误差值;根据误差值,更新所述权重值以及偏置值;使用更新后的权重值以及偏置值反向传播,以根据指令集合执行新一轮NPU片上计算。本申请提供的方法通过CPU、NPU协同,将LSTM网络中的运算分工,以解决CPU、GPU无法快速高效的计算LSTM网络的问题。
公开/授权文献
- CN116029332A 一种基于LSTM网络的片上微调方法及装置 公开/授权日:2023-04-28