c++ - 保存 caffe 模型 C++ 源代码？

Question

我正在尝试使用以下代码将 caffe 网络模型保存为 .caffemodel 格式。但是我遇到了段错误。

caffe::NetParameter net_param;
caffe_net_->ToProto(&net_param);
caffe::WriteProtoToBinaryFile(net_param, caffe_model_path);

其中 caffe_net_ 是经过训练的咖啡网络，其中包含 blob 参数。

Answer 1

不知道为什么会出现段错误。这正是您手动保存网络状态的方式。您是否尝试过单步执行并查看故障点？否则，只需在训练期间对其进行快照？