- 专利标题: 一种卷积神经网络模型加速训练方法、装置、电子设备及存储介质
-
申请号: CN202010947824.5申请日: 2020-09-10
-
公开(公告)号: CN112085166A公开(公告)日: 2020-12-15
- 发明人: 薛峰 , 张万友 , 夏炎
- 申请人: 江苏提米智能科技有限公司
- 申请人地址: 江苏省南京市浦口区江北新区星火路14号长峰大厦2号楼305室
- 专利权人: 江苏提米智能科技有限公司
- 当前专利权人: 江苏提米智能科技有限公司
- 当前专利权人地址: 江苏省南京市浦口区江北新区星火路14号长峰大厦2号楼305室
- 代理机构: 南京勤行知识产权代理事务所
- 代理商 尹英
- 主分类号: G06N3/04
- IPC分类号: G06N3/04 ; G06N3/08
摘要:
本发明公开了一种卷积神经网络模型加速训练方法、装置、电子设备及存储介质,属于机器学习技术领域。该方法通过读取训练数据到内存中,遍历所述文件路径下所有文件名,将文件分批次转化成二进制的样本数据;读取所述样本数据,将所述样本数据打包成mini‑batch张量;创建填充协议缓冲区域,将所述mini‑batch张量输入网络作为输入点,实时获取所述样本数据;创建数据流图节点队列,将所述样本数据填充入卷积神经网络中训练,能够缓解模型训练阶段的流水线读取数据的速度瓶颈,优化数据加速过程中的性能,提高卷积神经网络的训练速度。
公开/授权文献
- CN112085166B 一种卷积神经网络模型加速训练方法、装置、电子设备及存储介质 公开/授权日:2024-05-10