原创力文档- 1
- 0
- 约1.5万字
- 约 13页
- 2024-02-28 发布于四川
- 举报
本披露公开了一种用于加速异构图神经网络的大数据集训练的方法及相关产品。其中，前述的方法包括循环地执行以下步骤：基于中央处理单元CPU对包含异构图的大数据集进行采样处理，以得到小批量的训练集；基于所述CPU对所述训练集进行数据处理，以得到数据处理结果；以及基于图像处理单元GPU获取所述数据处理结果，并根据所述数据处理结果训练所述异构图神经网络。通过本披露的技术方案，可以将异构图神经网络训练过程所涉及的各个任务分配至CPU和GPU上执行，特别是将对GPU计算资源需求不大的采样和数据处理过程分配至CP
(19)国家知识产权局
(12)发明专利申请
(10)申请公布号CN117610616A
(43)申请公布日2024.02.27
(21)申请号202311464998.6
(22)申请日2023.11.06
(71)申请人上海处理器技术创新中心
地址2
文档评论（0）