-
公开(公告)号：CN117610616A
公开(公告)日：2024-02-27
申请号：CN202311464998.6
申请日：2023-11-06
Applicant: 上海处理器技术创新中心
IPC: G06N3/0464 , G06N3/08 , G06F9/50
Abstract: 本披露公开了一种用于加速异构图神经网络的大数据集训练的方法及相关产品。其中，前述的方法包括循环地执行以下步骤：基于中央处理单元CPU对包含异构图的大数据集进行采样处理，以得到小批量的训练集；基于所述CPU对所述训练集进行数据处理，以得到数据处理结果；以及基于图像处理单元GPU获取所述数据处理结果，并根据所述数据处理结果训练所述异构图神经网络。通过本披露的技术方案，可以将异构图神经网络训练过程所涉及的各个任务分配至CPU和GPU上执行，特别是将对GPU计算资源需求不大的采样和数据处理过程分配至CPU上执行，可以有效提高GPU利用率，从而加速整个训练过程。
Search Results
1
records
(took 0.02 seconds)
筛选
AND
AND
过滤
NOT
NOT
Scan to Join
