steflee / mpi-caffe

mpi-caffe
http://homes.soic.indiana.edu/steflee/mpi-caffe.html
Other
49 stars 21 forks source link

[This code is no longer supported and way behind the times at this point.]

mpi-caffe combines the modularity of the popular Caffe deep learning framework with the powerful Message Passing Interface (MPI) standard -- enabling simple, modular design of deep networks that span multiple machines!

Communication logic in mpi-caffe is abstracted to model layers which are included as part of the network architecture, allowing researchers to quickly experiment with model-distributed networks in environments ranging from a multiple GPU workstation to a large-scale cluster. With mpi-caffe it is easy to experiment with extremely large networks which don't fit into a single node's memory, link multiple full networks under a unified loss, or allow a set of networks to share some common computation.