snuspl / dolphin

14 stars 2 forks source link

Support batch processing of DNN #141

Closed beomyeol closed 8 years ago

beomyeol commented 8 years ago

In current implementation of dolphin-dnn, the neural network model is trained with only one input instance at each time. If we train the model with input batches, we can get better performance on computing activation values especially in fully connected layers since activations of multiple input instances are computed with one matrix multiplication. Also, in model partitioning, we can split input batches, and each evaluator in the same model takes these splits and computes activation values and gradients.

bgchun commented 8 years ago

Good job!