公开(公告)号：US20200272907A1

公开(公告)日：2020-08-27

申请号：US16748284

申请日：2020-01-21

Applicant: Huazhong University of Science and Technology

Inventor： Hai JIN ,  Xiaofei LIAO ,  Long ZHENG ,  Haikun LIU ,  Xi GE

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

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.