Caffe编译错误:blob大小超过INT_MAX
在使用Caffe进行编译时,有时会遇到错误提示"blob size exceeds INT_MAX"。这个错误通常是由于输入的数据太大,超出了INT_MAX的限制。为了解决这个问题,我们可以对Caffe的源代码进行一些修改。
下面是一个简单的例子,展示了如何修改Caffe的源代码以解决这个问题。请注意,以下代码仅用于说明目的,实际情况可能因版本差异而有所不同。
首先,找到Caffe的源代码目录,并进入到"src/caffe/proto"目录下,找到"caffe.proto"文件。打开该文件,我们需要添加一个新的字段来扩展INT_MAX的限制。
message BlobShape {
repeated int64 dim = 1 [packed = true];
}
message BlobProto {
optional BlobShape shape = 7;
optional float version = 1 [default = 0];
repeated float data = 2;
repeated double double_data = 8;
repeated int32 int32_data = 3;
repeated int64 int64_data = 4;
repeated bool bool_data = 5;
repeated int32 shape_data = 6;
}
上述代码中,我们添加了一个名为"shape"的新字段,它是一个类型为"BlobShape"的可选字段。
订阅专栏 解锁全文
2593

被折叠的 条评论
为什么被折叠?



