保存caffe Model C ++源代码?

时间:2015-09-18 07:17:52

标签: c++ deep-learning caffe

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

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

其中caffe_net_是经过训练的caffe网,wgich包含blob参数。

1 个答案:

答案 0 :(得分:0)

不知道为什么你会从那里得到一个seg错误。这正是您手动保存网络状态的方式。你试过踩过去看看它有什么不对吗?否则,只是在训练期间快照它?

相关问题