Invention Grant
- Patent Title: 一种基于LSTM网络的片上微调方法及装置
-
Application No.: CN202310147794.3Application Date: 2023-02-22
-
Publication No.: CN116029332BPublication Date: 2023-08-22
- Inventor: 林军 , 周李鑫 , 崔峻伟 , 王中风
- Applicant: 南京大学
- Applicant Address: 江苏省南京市栖霞区仙林大道163号
- Assignee: 南京大学
- Current Assignee: 南京大学
- Current Assignee Address: 江苏省南京市栖霞区仙林大道163号
- Agency: 北京弘权知识产权代理有限公司
- Agent 逯长明; 占园
- Main IPC: G06N3/0442
- IPC: G06N3/0442 ; G06N3/048 ; G06N3/084 ; G06N3/063

Abstract:
本申请提供一种基于LSTM网络的片上微调方法及装置,方法包括:根据LSTM网络模型的参数,生成计算参数;根据计算参数对输入值和权重值进行细粒度切分,以及生成指令集合;在神经网络处理器NPU读取指令集合;根据指令集合执行NPU片上计算,以生成中间值和LSTM网络输出值;将LSTM网络输出值传输至误差函数,并通过误差函数得到误差值;根据误差值,更新所述权重值以及偏置值;使用更新后的权重值以及偏置值反向传播,以根据指令集合执行新一轮NPU片上计算。本申请提供的方法通过CPU、NPU协同,将LSTM网络中的运算分工,以解决CPU、GPU无法快速高效的计算LSTM网络的问题。
Public/Granted literature
- CN116029332A 一种基于LSTM网络的片上微调方法及装置 Public/Granted day:2023-04-28
Information query