公开(公告)号：US11568268B2

公开(公告)日：2023-01-31

申请号：US16748284

申请日：2020-01-21

Applicant: Huazhong University of Science and Technology

Inventor： Hai Jin ,  Xiaofei Liao ,  Long Zheng ,  Haikun Liu ,  Xi Ge

IPC: G06N3/04 ,  G06N3/10 ,  G06F9/50 ,  G06N3/08

Abstract: A deep learning heterogeneous computing method based on layer-wide memory allocation, at least comprises steps of: traversing a neural network model so as to acquire a training operational sequence and a number of layers L thereof; calculating a memory room R1 required by data involved in operation at the ith layer of the neural network model under a double-buffer configuration, where 1≤i≤L; altering a layer structure of the ith layer and updating the training operational sequence; distributing all the data across a memory room of the CPU and the memory room of the GPU according to a data placement method; performing iterative computation at each said layer successively based on the training operational sequence so as to complete neural network training.