Method, apparatus and device for updating convolutional neural network using GPU cluster
Abstract:
An example method for updating convolutional neural network includes: obtaining a sample with a classification label; performing a first operation on the sample based on parameters of each layer of front-end network, to obtain a first operation result; performing a second operation on the sample based on the first operation result and the parameters of each layer of back-end network that the first GPU has, to obtain a second operation result; separately sending the first operation result to the other GPUs; receiving a third operation result obtained after each other GPU performs a third operation on the sample based on their parameters of each layer of back-end network and the first operation result; combining the second and third operation results to obtain a classification result; determining a prediction error based on the classification result and the classification label; and updating the convolutional neural network based on the prediction error.
Information query
Patent Agency Ranking
0/0