保存caffe模型的c++源代码

Saving caffe Model C++ source code?

本文关键字：c++ 源代码 模型 caffe 保存      更新时间：2023-10-16

我正在尝试使用以下代码将caffe net模型保存为. caffmodel格式。但我得到分割错误。

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

其中caffe_net_是一个训练过的caffe网络，其中包含blobs参数。

不知道为什么你会得到一个分段错误。这正是手动保存网络状态的方法。你有没有试着跨过去看看它在什么地方出错了?否则，只是在训练时拍快照?

相关文章：