Initialize the transformer rng in the base data layer

BVLC · Sep 3, 2014 · a08f111 · a08f111
1 parent 858ad41
commit a08f111
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/caffe/layers/base_data_layer.cpp b/src/caffe/layers/base_data_layer.cpp
@@ -46,6 +46,7 @@ void BaseDataLayer<Dtype>::LayerSetUp(const vector<Blob<Dtype>*>& bottom,
     data_mean_.Reshape(1, datum_channels_, datum_height_, datum_width_);
   }
   mean_ = data_mean_.cpu_data();
+  data_transformer_.InitRand();
 }
 
 template <typename Dtype>