BVLC / caffe

Caffe: a fast open framework for deep learning.
http://caffe.berkeleyvision.org/
Other
34.12k stars 18.69k forks source link

protobuf fatal exception in TENSORRT minist model optimizing #6366

Closed Arihe12th closed 6 years ago

Arihe12th commented 6 years ago

I am trying to run tensorrt python API code which given in this link https://docs.nvidia.com/deeplearning/sdk/tensorrt-api/topics/topics/workflows/caffe_to_tensorrt.html

that code is example of optimizing MNIST model using tensorrt in caffe I don"t find predefined mnist inside caffe/data/mnist folder so i used mnist.caffe,mnist.prototxt,mean.binaryproto form this github link https://github.com/DigitalGlobe/gbdx-caffe/tree/master/work/input/model.

while running the code I am getting this protobyf error

[libprotobuf FATAL /home/erisuser/p4sw/sw/gpgpu/MachineLearning/DIT/externals/protobuf/x86_64/10.0/include/google/protobuf/repeated_field.h:1078] CHECK failed: (index) < (currentsize): terminate called after throwing an instance of 'google_private::protobuf::FatalException' what(): CHECK failed: (index) < (currentsize): Aborted (core dumped)

can anyone help me to solve this. and it will more be help full is there demo on optimizing mnist build using caffe in tensorrt .

twmht commented 6 years ago

this should be posted to TensorRT forums (https://devtalk.nvidia.com/default/board/53/accelerated-computing/).