diff options
Diffstat (limited to 'src/caffe')
-rw-r--r-- | src/caffe/util/io.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/caffe/util/io.cpp b/src/caffe/util/io.cpp index ffdaeb12..6fa6e680 100644 --- a/src/caffe/util/io.cpp +++ b/src/caffe/util/io.cpp @@ -17,6 +17,8 @@ #include "caffe/proto/caffe.pb.h" #include "caffe/util/io.hpp" +const int kProtoReadBytesLimit = INT_MAX; // Max size of 2 GB minus 1 byte. + namespace caffe { using google::protobuf::io::FileInputStream; @@ -50,7 +52,7 @@ bool ReadProtoFromBinaryFile(const char* filename, Message* proto) { CHECK_NE(fd, -1) << "File not found: " << filename; ZeroCopyInputStream* raw_input = new FileInputStream(fd); CodedInputStream* coded_input = new CodedInputStream(raw_input); - coded_input->SetTotalBytesLimit(1073741824, 536870912); + coded_input->SetTotalBytesLimit(kProtoReadBytesLimit, 536870912); bool success = proto->ParseFromCodedStream(coded_input); |