Data Layers Parallel for Multi-GPU
authorRonghang Hu <huronghang@hotmail.com>
Wed, 12 Aug 2015 04:38:06 +0000 (21:38 -0700)
committerRonghang Hu <huronghang@hotmail.com>
Wed, 12 Aug 2015 17:51:45 +0000 (10:51 -0700)
commit0d34d5ba0fbdc09ac8f372cb581ccaec599f10bc
tree1d6aa3258483de57074730ba7e55e1fb5870e793
parent8771d0f4317fc0081d86b7637f5f5ceef5b92dfb
Data Layers Parallel for Multi-GPU

Allow data layers (and also PythonLayer when used as data layer) to be shared
among worker solver's training net, and also test net for future-proof if one
wants to do Multi-GPU testing. Data layers are locked during forward to ensure
sequential forward.
include/caffe/data_layers.hpp
include/caffe/layer.hpp
include/caffe/net.hpp
include/caffe/python_layer.hpp
include/caffe/solver.hpp
src/caffe/net.cpp
src/caffe/parallel.cpp
src/caffe/proto/caffe.proto
src/caffe/solver.cpp