Deep learning framework by BAIR
Created by
Yangqing Jia
Lead Developer
Evan Shelhamer
MemoryData
./include/caffe/layers/memory_data_layer.hpp
./src/caffe/layers/memory_data_layer.cpp
The memory data layer reads data directly from memory, without copying it. In order to use it, one must call MemoryDataLayer::Reset
(from C++) or Net.set_input_arrays
(from Python) in order to specify a source of contiguous data (as 4D row major array), which is read one batch-sized chunk at a time.
MemoryDataParameter memory_data_param
)./src/caffe/proto/caffe.proto
:batch_size
, channels
, height
, width
: specify the size of input chunks to read from memory