diff --git a/src/caffe/layers/base_data_layer.cpp b/src/caffe/layers/base_data_layer.cpp index 058e361ac59..9b1d55831a7 100644 --- a/src/caffe/layers/base_data_layer.cpp +++ b/src/caffe/layers/base_data_layer.cpp @@ -46,6 +46,7 @@ void BaseDataLayer::LayerSetUp(const vector*>& bottom, data_mean_.Reshape(1, datum_channels_, datum_height_, datum_width_); } mean_ = data_mean_.cpu_data(); + data_transformer_.InitRand(); } template