Huazhong University of Science and Technology

Xi Ge

Hai Jin

Xiaofei Liao

Long Zheng

Haikun Liu

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 Rrequired by data involved in operation at the ilayer of the neural network model under a double-buffer configuration, where 1&#x2264;i&#x2264;L; altering a layer structure of the ilayer 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.

Growing community of inventors

Hubei, China

Xi Ge