Allow using arrays with n_ * size_ > 2^31

uint64_t

size_t

diff --git a/include/caffe/data_layers.hpp b/include/caffe/data_layers.hpp
index 1542af7..e0d5a8a 100644 (file)
--- a/include/caffe/data_layers.hpp
+++ b/include/caffe/data_layers.hpp
@@ -281,7 +281,7 @@ class MemoryDataLayer : public BaseDataLayer<Dtype> {
   Dtype* data_;
   Dtype* labels_;
   int n_;
-  int pos_;
+  size_t pos_;
   Blob<Dtype> added_data_;
   Blob<Dtype> added_label_;
   bool has_new_data_;