我正在尝试使用c++在caffe框架中读取重量和偏见。这是我的代码
shared_ptr<Blob<float> >& weight = current_layer->blobs()[0];//for weights
shared_ptr<Blob<float> >& bias = current_layer->blobs()[1];//for bias
但是,如果对于某些模型,不存在或没有定义偏差,则通过细分错误来确定偏差。
那么哪个函数返回一个 bool(boolean) 值,该值指示存在偏见以及如何在c++中调用该函数?
最佳答案
从blobs
返回的current_layer->blobs()
存储在 std::vector
中,您可以使用其size
属性:
if (current_layer->blobs().size() > 1) {
shared_ptr<Blob<float> >& bias = current_layer->blobs()[1];//for bias
}
有关更多详细信息,请参见this similar answer以获取python接口(interface)。